AK牛牛

NLP

备战春招40天(25) 日常(4) 神经网络(11) 面试冲冲冲(3) 面试算法(24)

/ 注册

全部文章 / NLP （共1篇）

迁移学习-预训练模型的保存与加载

1，模型保存和读取： # 如果要保存最好的参数，使用: best_model_state = deepcopy(model.state_dict()) model_save_path = os.path.join('', 'model.pt') torch.save(model.state_dict...

2023-05-01

0 677