subtract_imagenet_mean_batch is not working

zhanghang1989 / PyTorch-Multi-Style-Transfer

Neural Style and MSG-Net

http://hangzh.com/PyTorch-Style-Transfer/

MIT License

977 stars 206 forks source link

Closed underfitting closed 7 years ago

underfitting commented 7 years ago

Hi,

I think the function utils.subtract_imagenet_mean_batch does not work with current code. You need to add "return batch" in the function and

utils.subtract_imagenet_mean_batch(xc)

should be changed to

xc = utils.subtract_imagenet_mean_batch(xc)

zhanghang1989 commented 7 years ago

Hi, thanks for pointing out the mistake! it is fixed in recent commit. 👍