- 博客(1)
- 收藏
- 关注
RSS订阅转载 PyTorch 预训练模型，保存，读取和更新模型参数以及多 GPU 训练模型（转载 极市平台）
PyTorch 预训练模型，保存，读取和更新模型参数以及多 GPU 训练模型转载 极市平台 9月10日目录1. PyTorch 预训练模型2.保存模型参数3.读取模型参数4. 冻结部分模型参数，进行 fine-tuning方法一：方法二：5. 模型训练与测试的设置训练时，应调用 model.train() ；测试时，应调用 model.eval()，以及 with torch.no_grad()...
2019-12-09 14:46:43
1901
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人