【最佳实践】对多个模型进行训练时的参数获取问题

最新推荐文章于 2024-06-16 00:39:17 发布

Sailist

最新推荐文章于 2024-06-16 00:39:17 发布

阅读量470

点赞数

分类专栏： Pytorch

本文链接：https://blog.csdn.net/sailist/article/details/103793892

版权

Pytorch 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

使用itertools中的chain方法将模型中的参数链接到一起：

optimizer = optim.Adam(params=chain(backbone_net.parameters(),
                                                linear_rot_net.parameters(),
                                                linear_classify_net.parameters()),
                                   lr=0.09,
                                   weight_decay=1e-4)