- 博客(1)
- 收藏
- 关注
原创 Pytorch预训练模型的使用
Pytorch预训练模型的使用加载预训练模型使用预训练模型的部分参数加载预训练模型state_dict = torch.load("H:/data/weights/checkpoints/vgg16-397923af.pth")model.load_state_dict(state_dict)使用预训练模型的部分参数方法一 ,将会带来全新的写作体验;new_state_dict = {}new_state_dict = { k: v for k, v in state_dict.item
2020-10-18 19:54:31 564
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人