pytorch/examples

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

Open

#391 aberto em 25 de jul. de 2018

Ver no GitHub
 (0 comments) (0 reactions) (0 assignees)Python (9.429 forks)batch import
good first issue

Métricas do repositório

Stars
 (21.634 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

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