From 8339f536f4c1b84da17d5b1fd9812f97aea8e52f Mon Sep 17 00:00:00 2001 From: yongjun Date: Mon, 16 Dec 2019 16:28:52 +0900 Subject: [PATCH] Fix version up error --- show_cls.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/show_cls.py b/show_cls.py index 6d054265e..7e831fb7b 100644 --- a/show_cls.py +++ b/show_cls.py @@ -51,4 +51,4 @@ pred_choice = pred.data.max(1)[1] correct = pred_choice.eq(target.data).cpu().sum() - print('i:%d loss: %f accuracy: %f' %(i, loss.data[0], correct/float(32))) + print('i:%d loss: %f accuracy: %f' %(i, loss.item(), correct/float(32)))