diff --git a/intermediate_source/char_rnn_classification_tutorial.py b/intermediate_source/char_rnn_classification_tutorial.py index af82583063f..e98d38723ad 100644 --- a/intermediate_source/char_rnn_classification_tutorial.py +++ b/intermediate_source/char_rnn_classification_tutorial.py @@ -332,7 +332,7 @@ def train(category_tensor, line_tensor): # Add parameters' gradients to their values, multiplied by learning rate for p in rnn.parameters(): - p.data.add_(-learning_rate, p.grad.data) + p.data.add_(p.grad.data, alpha=-learning_rate) return output, loss.item() diff --git a/intermediate_source/char_rnn_generation_tutorial.py b/intermediate_source/char_rnn_generation_tutorial.py index ec1d750ab26..a418e0abcd3 100644 --- a/intermediate_source/char_rnn_generation_tutorial.py +++ b/intermediate_source/char_rnn_generation_tutorial.py @@ -288,7 +288,7 @@ def train(category_tensor, input_line_tensor, target_line_tensor): loss.backward() for p in rnn.parameters(): - p.data.add_(-learning_rate, p.grad.data) + p.data.add_(p.grad.data, alpha=-learning_rate) return output, loss.item() / input_line_tensor.size(0)