fix #586 (#593)

lanpa · Jun 28, 2020 · 8f50866 · 8f50866
1 parent 022f060
commit 8f50866
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorboardX/utils.py b/tensorboardX/utils.py
@@ -80,7 +80,7 @@ def make_grid(I, ncols=8):
     W = I.shape[3]
     ncols = min(nimg, ncols)
     nrows = int(np.ceil(float(nimg) / ncols))
-    canvas = np.zeros((I.shape[1], H * nrows, W * ncols))
+    canvas = np.zeros((I.shape[1], H * nrows, W * ncols), dtype=I.dtype)
     i = 0
     for y in range(nrows):
         for x in range(ncols):