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)  编辑  收藏  举报