pytorch/examples

MNIST test_loader not being used as intended (wrong)!

Open

#756 opened on 2020年4月19日

GitHub で見る
 (3 comments) (0 reactions) (0 assignees)Python (21,634 stars) (9,429 forks)batch import
good first issue

説明

In the MNIST code, around line 57 (I think), the following code is wrong in my opinion for data, target in test_loader: It should be replaced by for idx,(data, target) in test_loader: The original code gives an error (int has no attribute .to(device)) as originally, data is actually assigned an integer (idx value) wrongly. I have verified this fix works. The code for training is correct but surprisingly there is a mistake in test code.

コントリビューターガイド