zhanghang1989 / PyTorch-Multi-Style-Transfer

Neural Style and MSG-Net
http://hangzh.com/PyTorch-Style-Transfer/
MIT License
976 stars 207 forks source link

Loaded samples aren't normalized and calculated loss is very large? #40

Open YuhuiMa opened 3 years ago

YuhuiMa commented 3 years ago

In the code, loaded samples seem only to substract mean but not to be normalized, which leads to very large loss. Why not normalize the samples?

zhanghang1989 commented 3 years ago

The VGG model I used was converted from Caffe, which was trained without std normalization.