There was an error while loading. Please reload this page.
1 parent 2d0f1c4 commit 08be28eCopy full SHA for 08be28e
fast_neural_style/neural_style/utils.py
@@ -39,5 +39,5 @@ def normalize_batch(batch):
39
std[:, 2, :, :] = 0.225
40
batch = torch.div(batch, 255.0)
41
batch -= Variable(mean)
42
- batch /= Variable(std)
+ batch = batch / Variable(std)
43
return batch
0 commit comments