原文:解决torch.save时遇到的UserWarning

在用PyTorch保存模型时,常常会遇到UserWarning: Couldn t retrieve source code for container of type Net. It won t be checked for correctness upon loading. type obj. name . It won t be checked ,可参考这篇知乎文章解决这类警告。 ...

2020-08-12 13:08 0 1220 推荐指数:

查看详情

torch.save torch.load 加载和保存模型

https://pytorch123.com/ThirdSection/SaveModel/ 这个链接非常的详细! 1、#保存整个网络 torch.save(net, PATH) # 保存网络中的参数, 速度快,占空间少 torch.save(net.state_dict(),PATH) #------------------------------------------------- ...

Wed Mar 09 01:57:00 CST 2022 0 1231
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM