Fix for Total Variational Loss

xunhuang1995 / AdaIN-style

Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization

https://arxiv.org/abs/1703.06868

MIT License

1.45k stars 193 forks source link

Closed ArturoDeza closed 5 years ago

ArturoDeza commented 5 years ago

Change at line 266: local tvLoss = criterion.net:get(2).loss to: local tvLoss = criterion.net:get(1).loss