This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow
Input data: Two synthetic regression tasks generated according to: Ma et al. (KDD 2018) "Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts"
Network: One shared layer Two task specific towers
Optimizer: Adam
Regularization: Dropout