pytorch/examples

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

Open

#391 创建于 2018年7月25日

在 GitHub 查看
 (0 评论) (0 反应) (0 负责人)Python (9,429 fork)batch import
good first issue

仓库指标

Star
 (21,634 star)
PR 合并指标
 (30 天内没有已合并 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

贡献者指南