2023年7月31日
模型权重保存、加载、冻结(pytorch)
摘要： 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 湾仔码农 阅读(1236) 评论(0) 推荐(0) 编辑