search

udacity / deep-learning-v2-pytorch

Projects and exercises for the latest Deep Learning ND program https://www.udacity.com/course/deep-learning-nanodegree--nd101
MIT License
5.31k stars 5.33k forks source link

Normalization is Transfer Learning #203

Closed puhach closed 3 years ago

puhach commented 5 years ago

It looks like normalization step is missing in Transfer Learning (Convolution Networks)

According to torchvision documentation, the the input images have to be loaded in to a range of [0, 1] and then normalized using mean = [0.485, 0.456, 0.406] and std = [0.229, 0.224, 0.225].

soni-ratnesh commented 4 years ago

If it’s not implemented I can implement and create a pull request.

abhiojha8 commented 3 years ago

Happy to review and approve a PR for this. Thanks :)