torch.save()用法

torch.save()用法：保存模型参数html filepath = os.path.join(args.save_path, 'checkpoint_model_epoch_{}.pth.tar'.format(epoch)) #最终参数模型 torch.save(state, filepath) 参考连接： https://www.cnblogs.com/qinduanyinghua/p/

>>阅读原文<<