From afd78073df5eca16c326201814eef3aba11704e7 Mon Sep 17 00:00:00 2001 From: Hongfei JIANG Date: Fri, 6 Sep 2019 11:27:28 +0800 Subject: [PATCH] Update train.py Use tensor.item() to convert a 0-dim tensor to a Python number --- char-rnn-classification/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/char-rnn-classification/train.py b/char-rnn-classification/train.py index 5831980..9ad2960 100644 --- a/char-rnn-classification/train.py +++ b/char-rnn-classification/train.py @@ -42,7 +42,7 @@ def train(category_tensor, line_tensor): optimizer.step() - return output, loss.data[0] + return output, loss.item() # Keep track of losses for plotting current_loss = 0