A custom Huggingface trainer which supports logging auxiliary losses returned by your model
Primary LanguagePythonMIT LicenseMIT