pytorch
请设置你的昵称
这个作者很懒,什么都没留下…
展开
-
pytorch的一些用法
日志:https://juejin.im/post/5bc2bd3a5188255c94465d31 lr:optimizer用param_groups来管理参数[optimizer.param_groups],其中保存了参数组和学习率动量等,以dict的形式保存,如optimizer.param_groups['lr'] model.load_state_dict:有...原创 2019-02-21 11:04:50 · 152 阅读 · 0 评论 -
Pytorch 模型fusion
模型A和模型B,两个模型共享backbone,现需要将两模型共享backbone参数backbone为resnet18步骤:1.训练单独的A模型2.load第一步的模型,将backbone中的某些层之前lr调小,之后的lr不变,训练B模型3.load第二步的模型,训练A模型,同时需要固定第二步中某些层的参数trick:1.第2、3步load模型的时候,会有一...原创 2019-04-18 14:57:06 · 1070 阅读 · 0 评论