pytorch/examples

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

Aperta

#391 aperta il 25 lug 2018

 (0 commenti) (0 reazioni) (0 assegnatari)Python (9429 fork)batch import
good first issue

Metriche repository

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

Descrizione

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

Guida contributor