Fix numpy issue in vis.py

2021-05-20 17:33:19 +02:00
parent 16dc3cf4eb
commit de63eaf15a
1 changed files with 1 additions and 1 deletions
--- a/prototorch/models/vis.py
+++ b/prototorch/models/vis.py
@@ -42,7 +42,7 @@ class Vis2DAbstract(pl.Callback):
            x, y = data

        if flatten_data:
-            x = x.view(len(data), -1)
+            x = x.reshape(len(x), -1)

        self.x_train = x
        self.y_train = y