pytorch/examples

InceptionV3 cannot work!

Aperta

#362 aperta il 27 mag 2018

Vedi su GitHubGuida contributor
 (8 commenti) (0 reazioni) (0 assegnatari)Python (9429 fork)batch import
help wantedvision

Metriche repository

Star
 (21.634 stelle)
Metriche merge PR
 (Metriche PR in attesa)

Descrizione

python main.py -a inception_v3 ./imagenet/cat2dog --batch-size 16 --print-freq 1 --pretrained; => using pre-trained model 'inception_v3' Traceback (most recent call last): File "main.py", line 314, in main() File "main.py", line 157, in main train(train_loader, model, criterion, optimizer, epoch) File "main.py", line 189, in train target = target.cuda(non_blocking=True) TypeError: _cuda() got an unexpected keyword argument 'non_blocking'

Guida contributor

Direzione di ricerca
Controlla la versione di PyTorch; l'argomento non blocking per .cuda() è stato introdotto in PyTorch 0.4.0. Aggiorna PyTorch a >=0.4.0 o rimuovi l'argomento non blocking=True dalla chiamata target.cuda().
Tech stack
pythonpytorch
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.
Meno di 1 ora
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
GitPython
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
75