1. 保存模型：torch.save(model.state_dict(), PATH)
加载模型：model.load_state_dict(torch.load(PATH))加载模型：m
阅读背景：
Pytorch预训练模型加载
来源：互联网
分享到：
