pytorch/examples

MNIST test_loader not being used as intended (wrong)!

Open

Aperta il 19 apr 2020

Vedi su GitHub
 (3 commenti) (0 reazioni) (0 assegnatari)Python (21.634 star) (9429 fork)batch import
good first issue

Descrizione

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.

Guida contributor

Tech stack
python
Dominio
machine learning
Tipo issue
bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
under 1 hour
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
stale
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
clear
Prerequisiti
PythonBasic PyTorch
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
80
Direzione di ricerca
The issue is in the MNIST example code in the pytorch/examples repository, specifically in the test loop around line 57 of the mnist/main.py file. The bug is that `test loader` is iterated incorrectly: `for data, target in test loader` should be `for idx,(data, target) in test loader`. The fix is straightforward and confirmed by the issue author. To contribute, locate the file, apply the change, and submit a pull request.