Transformers4Rec icon indicating copy to clipboard operation
Transformers4Rec copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA-Merlin

[FEA] Extend HuggingFace Trainer class for multi-task learning support

Open gabrielspmoreira opened this issue 3 years ago • 1 comments

🚀 Feature request

Motivation

Your contribution

gabrielspmoreira avatar Oct 04 '21 13:10 gabrielspmoreira

Does Transformers4Rec currently support multi-task learning? That is predicting multiple tasks (brand and product_id) on the same data? @rnyak

Ahanmr avatar May 10 '22 14:05 Ahanmr