全部回复
LV.3
#1
doraiba2022
回复
您好,pytorch可以支持的最高版本是1.3。如果移植高版本pytorch对应的模型,可以在保存模型时设置参数
_use_new_zipfile_serialization=False
,即:
torch.save(model.state_dict(), model_cp,_use_new_zipfile_serialization=False)
2021-10-27 20:28:26
0
|
回复
请
登录
后评论