Skip to content

Commit 769cff9

Browse files
akjalokNicolasHugSvetlana Karslioglu
authored
Fix the loss initialization in intermediate_source/char_rnn_generation_tutorial.py (#2380)
* changed the loss init to make it less confusing --------- Co-authored-by: Nicolas Hug <contact@nicolas-hug.com> Co-authored-by: Svetlana Karslioglu <svekars@fb.com>
1 parent b966c1f commit 769cff9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intermediate_source/char_rnn_generation_tutorial.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,7 @@ def train(category_tensor, input_line_tensor, target_line_tensor):
278278

279279
rnn.zero_grad()
280280

281-
loss = 0
281+
loss = torch.Tensor([0]) # you can also just simply use ``loss = 0``
282282

283283
for i in range(input_line_tensor.size(0)):
284284
output, hidden = rnn(category_tensor, input_line_tensor[i], hidden)

0 commit comments

Comments
 (0)