Gamus的博客

转载 【无标题】

pytorch保存模型等相关参数，利用torch.save()，以及读取保存之后的文件本文分为两部分，第一部分讲如何保存模型参数，优化器参数等等，第二部分则讲如何读取。假设网络为model = Net(), optimizer = optim.Adam(model.parameters(), lr=args.lr), 假设在某个epoch，我们要保存模型参数，优化器参数以及epoch一、先建立一个字典，保存三个参数：state = {‘net’:model.state_dict(), ‘opt