Open drpetershaw opened 5 years ago
Further digging makes me believe that it may be an issue with Pytorch compatibility as i have Pytorch Version=1.0.0
I tried to run this code in 1.0.0 with the --save_fig option and the same error occurred.
It seems that the error occurred because multiplication between Numpy array and Pytorch tensor is deprecated in Pytorch 1.0.0. I have fixed the error so check it out.
When i use:
!python /content/drive/My\ Drive/PlasmaSimAI/RNN/1_train_predictor.py --data ecg --filename chfdbchf15.pkl --save_fig
in Google colab the following error occurs:
When you remove --save_fig the code works as expected. Is this a bug or am i doing something wrong?