/multi_task_utils

A simple python package for multi-task training: wrappers for pytorch DataLoader and pytorch-lightning DataModule

Primary LanguagePythonMIT LicenseMIT

Stargazers