2021-01-25

本文探讨了如何将ResNet18的输出分割并输入到另一个网络中进行联合训练的方法。特别关注了如何处理反向传播(loss backward)的问题以及优化器的设置。

两个神经网络怎么联合训练?

比如说第一个为resnet18,输出为512维向量,我把它分成16个32维的数据,每一个数据再进行处理后(比如说裁剪中间16维)输入到第二个网络,loss怎么backward?优化器是不是设置为第二个网络的的参数,如果retain_graph不设置为true,就会导致loss无法backward,若设置为true则会计算爆炸,OOM。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值