- 博客(4)
- 收藏
- 关注
RSS订阅原创 assert all(map(lambda i: i.is_cuda, inputs)) AssertionError
assert all(map(lambda i: i.is_cuda, inputs)) AssertionError 报错，原因是pytorch在服务器上跑使用了多个gpu,暂时不知道怎么用多个gpu,但是为了跑通加入了指定单一gpu的代码： import os os.environ[“CUDA_VISIBLE_DEVICES”] = “1” 之后就不再报错了 欢迎留言，提出如何使用多个gpu而不会报错的方法 ...
2020-05-23 15:02:57
1862
4
原创 RuntimeError: CUDA out of memory.Tried to allocate 20.00 MiB GPU 0 ,3.94 GiB total capacity 3.36 G
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 3.94 GiB total capacity; 3.36 GiB already allocated; 13.06 MiB free; 78.58 MiB cached) 减小batch_size, 致敬这位老哥
2020-05-22 16:04:49
8767
7
原创 相对路径报错
相对路径报错 绝对路径可行，相对路径报错，提示找不到 从voc2ssd.py找到Annotations, 开始没有改，之前的路径为=r“./VOC2007/Annotations” python找不到路径，最后也不做其他判断了，利用os.path.relpath（“绝对路径”）找到相对路径 ...
2020-05-22 11:36:37
648
原创 pytorch报错：Target is out of bounds
pytorch报错：Target is out of bounds 查看网络的最后输出，softmax的输出节点数是否等于所有的标签数
2020-04-14 00:01:28
18532
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人