From f49cbe5a12e96dd656bab3fe919d0a0d02dc484a Mon Sep 17 00:00:00 2001 From: Hust_XianYang Date: Tue, 22 Oct 2019 11:19:35 +0800 Subject: [PATCH] Update omniglot_train_few_shot.py update the pytorch about "data[0]" to item() --- omniglot/omniglot_train_few_shot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/omniglot/omniglot_train_few_shot.py b/omniglot/omniglot_train_few_shot.py index 58e0b529..337b048c 100644 --- a/omniglot/omniglot_train_few_shot.py +++ b/omniglot/omniglot_train_few_shot.py @@ -203,7 +203,7 @@ def main(): relation_network_optim.step() if (episode+1)%100 == 0: - print("episode:",episode+1,"loss",loss.data[0]) + print("episode:",episode+1,"loss",loss.item()) if (episode+1)%5000 == 0: