2021年4月8日

pytorch训练多分支网络

摘要： 一个多分支输出网络（一个Encoder，多个Decoder） 我们期望每个分支的损失L_i分别对各自的参数进行优化，而共享参数部分采用各分支损失之和Sum（L_i）进行优化。 在pytorch中是默认支持这种操作的，也就是我们可以分别计算出各分支的loss，然后直接把他们相加即可。（参考上面pyto 阅读全文

posted @ 2021-04-08 23:00 那抹阳光1994 阅读(8694) 评论(2) 推荐(1)

导航