--resume fails after 1 epoch with Pytorch 1.0 release

Métriques du dépôt

Stars

21 634 stars (21 634 stars)

Métriques de merge PR

Aucune PR mergée en 30 j (Aucune PR mergée en 30 j)

Description

Traceback (most recent call last):
  File "main.py", line 398, in <module>
    main()
  File "main.py", line 110, in main
    mp.spawn(main_worker, nprocs=ngpus_per_node, args=(ngpus_per_node, args))
  File "/home/tools/anaconda3-5.3/lib/python3.7/site-packages/torch/multiprocessing/spawn.py", line 167, in spawn
    while not spawn_context.join():
  File "/home/tools/anaconda3-5.3/lib/python3.7/site-packages/torch/multiprocessing/spawn.py", line 114, in join
    raise Exception(msg)
Exception:

-- Process 1 terminated with the following error:
Traceback (most recent call last):
  File "/home/tools/anaconda3-5.3/lib/python3.7/site-packages/torch/multiprocessing/spawn.py", line 19, in _wrap
    fn(i, *args)
  File "/space8T/mdflickner/pytorch/examples/imagenet/main.py", line 241, in main_worker
    is_best = acc1 > best_acc1
RuntimeError: arguments are located on different GPUs at /pytorch/aten/src/THC/generic/THCTensorMathCompareT.cu:15

Guide contributeur

Direction de recherche

Inspectez la fonction main worker dans main.py (ligne 241), où la comparaison entre best acc1 et acc1 provoque une RuntimeError en raison de placements GPU différents. Tracez comment best acc1 est initialisé et mis à jour au fil des époques, et assurez vous que les deux tenseurs sont déplacés sur le même appareil. Passez également en revue la logique resume pour vérifier que les state dicts sont chargés sur le bon appareil. L'erreur est reproductible avec resnet50/18 sur PyTorch 1.0.

Stack technique

pythonpytorch

Domaine

backendmachine 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.

3

Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.

Une demi journée

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.

Investigation requise

Prérequis

PythonPyTorch

Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.

40