摘要:
对比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)
浙公网安备 33010602011771号