2021
07-22
07-22
使用Pytorch训练two-head网络的操作
之前有写过一篇如何使用Pytorch实现two-head(多输出)模型在那篇文章里,基本把two-head网络以及构建讲清楚了(如果不清楚请先移步至那一篇博文)。但是我后来发现之前的训练方法貌似有些问题。以前的训练方法:之前是把两个head分开进行训练的,因此每一轮训练先要对一个batch的数据进行划分,然后再分别训练两个头。代码如下:f_out_y0,_=net(x0)_,f_out_y1=net(x1)#实例化损失函数...
继续阅读 >