Skip to content

Commit caea9a3

Browse files
committed
update how add_ method is used in char_rnn_* tuutorials to avoid warnings
1 parent f557ee0 commit caea9a3

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

intermediate_source/char_rnn_classification_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -332,7 +332,7 @@ def train(category_tensor, line_tensor):
332332

333333
# Add parameters' gradients to their values, multiplied by learning rate
334334
for p in rnn.parameters():
335-
p.data.add_(-learning_rate, p.grad.data)
335+
p.data.add_(p.grad.data, alpha=-learning_rate)
336336

337337
return output, loss.item()
338338

intermediate_source/char_rnn_generation_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,7 @@ def train(category_tensor, input_line_tensor, target_line_tensor):
288288
loss.backward()
289289

290290
for p in rnn.parameters():
291-
p.data.add_(-learning_rate, p.grad.data)
291+
p.data.add_(p.grad.data, alpha=-learning_rate)
292292

293293
return output, loss.item() / input_line_tensor.size(0)
294294

0 commit comments

Comments
 (0)