julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix numpy issue in vis.py

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-05-20 17:33:19 +02:00
parent 16dc3cf4eb
commit de63eaf15a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/models/vis.py
View File

@ -42,7 +42,7 @@ class Vis2DAbstract(pl.Callback):
x, y = data x, y = data
if flatten_data: if flatten_data:
x = x.view(len(data), -1) x = x.reshape(len(x), -1)
self.x_train = x self.x_train = x
self.y_train = y self.y_train = y