pytorch/examples

InceptionV3 cannot work!

Open

#362 ouverte le 27 mai 2018

Voir sur GitHub
 (8 commentaires) (0 réactions) (0 assignés)Python (9 429 forks)batch import
help wantedvision

Métriques du dépôt

Stars
 (21 634 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

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'

Guide contributeur

Direction de recherche
Vérifiez la version de PyTorch ; l'argument non blocking pour .cuda() a été introduit dans PyTorch 0.4.0. Mettez à jour PyTorch vers >=0.4.0 ou supprimez l'argument non blocking=True de l'appel target.cuda().
Stack technique
pythonpytorch
Domaine
machine learning
Type d'issue
Bug
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
Moins d'une heure
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
GitPython
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
75