Pytorch – 模型断点续训,optimizer.step()报错:RuntimeError Expected all tensors to be on the same device, but found cuda:0
1 模型断点续训,optimizer.step()报错:RuntimeError Expected all tensors to be on the same device, but found cuda:0 Pytroch在实现断点续训功能时,在保存模型文件时,需要同时保存model、optimi…
- Pytorch
- 2023-05-08