Skip to content

Commit 2a34c3c

Browse files
committed
Fix neural_style_tutorial
Signed-off-by: Onur Berk Töre <onurberk_t@hotmail.com>
1 parent 671960c commit 2a34c3c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

advanced_source/neural_style_tutorial.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@
107107
def image_loader(image_name):
108108
image = Image.open(image_name)
109109
# fake batch dimension required to fit network's input dimensions
110+
image = loader(image).unsqueeze(0)
110111
return image.to(device, torch.float)
111112

112113

0 commit comments

Comments
 (0)