pytorch/examples

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

Open

#391 opened on Jul 25, 2018

View on GitHub
 (0 comments) (0 reactions) (0 assignees)Python (9,429 forks)batch import
good first issue

Repository metrics

Stars
 (21,634 stars)
PR merge metrics
 (No merged PRs in 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

Contributor guide