多卡训练报错：assert all(map(lambda i: i.is_cuda, inputs))

onion_rain

于 2020-07-08 20:10:57 发布

阅读量1.2k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch

本文链接：https://blog.youkuaiyun.com/onion_rain/article/details/107112162

现象

多卡训练模型时候出错：

training...                                                                                                                                                                                                                          [13/1964]Train    0:   0%|                                                                    | 0/5334 [00:20<?, ?it/s]
Traceback (most recent call last):
  File "train.py", line 221, in <modu