摘要: 对比tf来说简单了好几个量级。。 1. 先建立一个字典,保存三个参数: # 只保存了参数 state = {‘net':model.state_dict(), 'optimizer':optimizer.state_dict(), 'epoch':epoch}torch.save(state, mo 阅读全文
posted @ 2020-11-27 15:09 小小马进阶笔记 阅读(411) 评论(0) 推荐(0)