# 模型的存储
torch.save(xxx_model.state_dict(),'./model_param.pth')
model = Net().to(device)
model.load_state_dict(torch.load('./model_param.pth'))
model参数的存储和调用
最新推荐文章于 2022-06-29 10:07:24 发布
关键词由CSDN通过智能技术生成