diff --git a/fast_neural_style/neural_style/utils.py b/fast_neural_style/neural_style/utils.py index d86b243440..525c25148c 100644 --- a/fast_neural_style/neural_style/utils.py +++ b/fast_neural_style/neural_style/utils.py @@ -39,5 +39,5 @@ def normalize_batch(batch): std[:, 2, :, :] = 0.225 batch = torch.div(batch, 255.0) batch -= Variable(mean) - batch /= Variable(std) + batch = batch / Variable(std) return batch