BBYR Achieve
返回信息流
这是一条镜像帖。来源：北邮人论坛 / ml-dm / #38475同步于 2022/6/22
该镜像源已超过 30 天没有更新，可能在源站已被删除。
ML_DM机器人发帖

tf多任务学习demo怎么写

zyu1998
2022/6/22镜像同步4 回复
我写了两个loss，但因为学习率不同不能直接加在一起训练，但基于estimator进行训练时返回实例tf.estimator.EstimatorSpec(model, loss, train_op)里得传一个loss，有好心人知道这个训练demo怎么写吗[ema1]
订阅后，新回复会通过你的通知中心匿名送达。
4 条回复
zyu1998机器人#1 · 2022/6/22
我目前只能让每一个batch随机选一个loss进行训练，但感觉不太对劲
cloudsquare机器人#2 · 2022/6/22
不能写一个损失函数的类把两个损失函数封在这里面吗？
ykt机器人#3 · 2022/6/23
一个参考：https://mp.weixin.qq.com/s/iAwGIipLRyOpdKTDRufgAg
zyu1998机器人#4 · 2022/6/23
主要两个loss对应参数学习率不同，不好和在一起 【 在 cloudsquare 的大作中提到: 】 : 不能写一个损失函数的类把两个损失函数封在这里面吗？