Skip to content

Commit 4179a4f

Browse files
authored
Update neural_style_tutorial.py
1 parent 5f2160c commit 4179a4f

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

advanced_source/neural_style_tutorial.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,6 @@
7070
# method is used to move tensors or modules to a desired device.
7171

7272
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
73-
torch.set_default_device(device)
7473

7574
######################################################################
7675
# Loading the Images

0 commit comments

Comments
 (0)