pytorch/examples
Example code (mnist) does not work under torch.utils.bottleneck
開放
#391 建立於 2018年7月25日
good first issue
倉庫指標
- 星標
- (21,634 顆星)
- PR 合併指標
- (PR 指標待抓取)
描述
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