摘要: 1. 保存整个网络 torch.save(net, PATH) model = torch.load(PATH) 2. 保存网络中的参数(速度快,占空间小) torch.save(net.state_dict(),PATH) model_dict = model.load_state_dict(to 阅读全文
posted @ 2023-07-31 21:35 湾仔码农 阅读(2071) 评论(0) 推荐(0) 编辑