search

udacity / DL_PyTorch

Code for the Deep Learning with PyTorch lesson
MIT License
361 stars 370 forks source link

fix: different syntax for PyTorch 1.x for transforms.Normalize #12

Open MohannadEhabBarakat opened 4 years ago

MohannadEhabBarakat commented 4 years ago

the original code is not working with PyTorch 1.x. So, I have updated the parameters of transforms.Normalize. The new code is working fine for me. I didn't test the rest of the code just the beginning of the notebook.

MohannadEhabBarakat commented 4 years ago

This is my first pull request so if I did something wrong or not following the best practice please let me know.