pytorch/examples

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

開放

#391 建立於 2018年7月25日

 (0 則留言) (0 個反應) (0 位負責人)Python (9,429 個分叉)batch import
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

貢獻者指南