Skip to content

Commit 7240041

Browse files
committed
adding device default to cpu since some commands involve a set device
1 parent 52374e0 commit 7240041

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

intermediate_source/char_rnn_classification_tutorial.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@
6666
# limit the RNN input layers. This is accomplished by converting Unicode strings to ASCII and allowing only a small set of allowed characters.
6767

6868
import torch
69+
device = torch.device('cpu')
6970
import string
7071
import unicodedata
7172

0 commit comments

Comments
 (0)