保存模型:
torch.save(model, 'model.pth')
加载模型:
model = torch.load('model.pth')
上一篇 1070 结绳 (25 分)
下一篇 leetcode剑之offer之第一个只出现一次的字符(十二)
版权所有 (c)2021-2022 MSHXW.COM
ICP备案号:晋ICP备2021003244-6号