- 博客(5)
- 收藏
- 关注
原创 【CUDA error: invalid device ordinal错误解决】
由于在训练时使用DataParallel,且torch.save使用如下代码保存。将之改为保存模型的参数(注意:必须使用。train.py(未验证是否可行)
2023-06-01 20:34:14 1524 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
由于在训练时使用DataParallel,且torch.save使用如下代码保存。将之改为保存模型的参数(注意:必须使用。train.py(未验证是否可行)
2023-06-01 20:34:14 1524 1
TA创建的收藏夹 TA关注的收藏夹
TA关注的人