Skip to content

Commit 08be28e

Browse files
twairballsoumith
authored andcommitted
This PR fixes error raised during element-wise variable division. (pytorch#182)
- Fixes pytorch#181
1 parent 2d0f1c4 commit 08be28e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fast_neural_style/neural_style/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,5 +39,5 @@ def normalize_batch(batch):
3939
std[:, 2, :, :] = 0.225
4040
batch = torch.div(batch, 255.0)
4141
batch -= Variable(mean)
42-
batch /= Variable(std)
42+
batch = batch / Variable(std)
4343
return batch

0 commit comments

Comments
 (0)