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/

>>阅读原文<<