search

easezyc / Multitask-Recommendation-Library

MTReclib provides a PyTorch implementation of multi-task recommendation models and common datasets.
MIT License
282 stars 45 forks source link

多任务loss的权重 #3

Closed Gary828 closed 2 years ago

Gary828 commented 2 years ago

您好，请问下您的代码中有实现ple中学习不同任务loss的权重吗？谢谢

easezyc commented 2 years ago

我没有实现loss的权重，这块可能在使用的时候需要自己加一下，这里主要是集成了不同多任务模型的框架，里面还有些小的细节不方便统一框架的就没有实现，比如AITM的校准器。

Gary828 commented 2 years ago

嗯嗯，感谢您的回复