pytorch/examples

Example code (mnist) does not work under torch.utils.bottleneck

Open

#391 ouverte le 25 juil. 2018

Voir sur GitHub
 (0 commentaires) (0 réactions) (0 assignés)Python (9 429 forks)batch import
good first issue

Métriques du dépôt

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

Description

Example code (mnist) does not work under torch.utils.bottleneck. This is solved by changing the value num_workers from 1 to 0. It needs to check other examples also.

Reference https://github.com/pytorch/pytorch/issues/6313 https://discuss.pytorch.org/t/cannot-measure-cuda-usage-on-mnist-example/21600/2

Guide contributeur