(#3) Defense: Cast EagerTensor to numpy array

betarixm · May 2, 2022 · 81c992c · 81c992c
1 parent 45744ce
commit 81c992c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/utils/logging.py b/src/utils/logging.py
@@ -8,7 +8,9 @@ def concat_batch_images(batch_images: np.array):
         np.concatenate(
             [
                 np.concatenate(i, axis=1)
-                for i in batch_images.reshape(-1, width, *batch_images.shape[1:])
+                for i in np.array(batch_images).reshape(
+                    -1, width, *batch_images.shape[1:]
+                )
             ],
             axis=0,
         ),