Pytorch版本问题
将
param_dict = torch.load(trained_path)
改为
param_dict = torch.load(trained_path).state_dict()
即可
上一篇 Linux 设置Docker容器开机自启动,Dokcer容器常用命令总结。
下一篇 go FTP 文件传输
版权所有 (c)2021-2022 MSHXW.COM
ICP备案号:晋ICP备2021003244-6号