fix shape bug for pytorch 0.4

hunkim / PyTorchZeroToAll

Simple PyTorch Tutorials Zero to ALL!

http://bit.ly/PyTorchZeroAll

3.89k stars 1.2k forks source link

Open ShuvenduRoy opened 6 years ago

ShuvenduRoy commented 6 years ago

There was a bug for pytorch 0.4 and a warning for 0.5 it will cause error