2020年6月30日
Pytorch加载预训练模型的坑
摘要： 保存模型： def save(model, model_path): torch.save(model.state_dict(), model_path) 加载模型： def load(model, model_path): model.load_state_dict(torch.load(mode 阅读全文
posted @ 2020-06-30 13:25 嶙羽 阅读(2273) 评论(0) 推荐(0) 编辑