pytorch/examples

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

Aberta

#391 aberto em 25 de jul. de 2018

 (0 comentário) (0 reação) (0 responsável)Python (9.429 forks)batch import
good first issue

Métricas do repositório

Stars
 (21.634 estrelas)
Métricas de merge de PR
 (Métricas PR pendentes)

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

Guia do colaborador