magicboom的博客

1.torch.save 1. 先建立一个字典，保存三个参数: state = {‘net':model.state_dict(), 'optimizer':optimizer.state_dict(), 'epoch':epoch} 2.调用torch.save():其中dir表示保存文件的绝对路径+保存文件名，如'/home/qinying/Desktop/modelpara.pth' torch.save(state, dir) 3.当你想恢复某一阶段的训练（或者进行测试）时，那.