check if cfg_early_exit_loss has curriculum

pytorch · Dec 1, 2024 · 40b7987 · 40b7987
1 parent e51419c
commit 40b7987
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recipes/dev/early_exit_finetune_distributed.py b/recipes/dev/early_exit_finetune_distributed.py
@@ -710,7 +710,7 @@ def _setup_early_exit_loss(
             train_last_layer = cfg_early_exit_loss.get("include_last_layer", True)
             verbose = cfg_early_exit_loss.get("verbose", False)
 
-            if cfg_early_exit_loss.curriculum:
+            if cfg_early_exit_loss.get("curriculum", None):
                 early_exit_loss_curriculum = setup_early_exit_loss_curriculum(
                     early_exit_curriculum=cfg_early_exit_loss.curriculum,
                     do_output_hidden_states=do_output_hidden_states,