分类：two-head模型

2021
07-22

使用Pytorch实现two-head(多输出)模型的操作

如何使用Pytorch实现two-head（多输出）模型1.two-head模型定义先放一张我要实现的模型结构图：如上图，就是一个two-head模型，也是一个但输入多输出模型。该模型的特点是输入一个x和一个t，h0和h1中只有一个会输出，所以可能这不算是一个典型的多输出模型。2.实现所遇到的困难一开始的想法：这不是很简单嘛，做一个判断不就完了，t=0时模型为前半段加h0，t=1时模型为前半段加h1。但实现的时候傻眼了，发现在真正前向传播的时候t... 继续阅读 >