Skip to content

feat: add a JSD option to the distillation loss - #4026

Open
kashif wants to merge 4 commits into
NVIDIA-NeMo:mainfrom
kashif:feat/distillation-jsd-loss
Open

kashif wants to merge 4 commits into
NVIDIA-NeMo:mainfrom
kashif:feat/distillation-jsd-loss