debug

alexandrainst · Sep 27, 2024 · 1202776 · 1202776
1 parent 4e30fc7
commit 1202776
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/coral/finetune.py b/src/coral/finetune.py
@@ -42,6 +42,8 @@ def finetune(config: DictConfig) -> None:
     if "val" not in dataset and is_main_process:
         logger.info("No validation set found. Disabling early stopping.")
 
+    breakpoint()
+
     trainer = model_setup.load_trainer_class()(
         model=model,
         data_collator=model_setup.load_data_collator(),
@@ -63,7 +65,6 @@ def finetune(config: DictConfig) -> None:
 
     # if hasattr(config.model, "use_decoder") and config.model.use_decoder:
     #     train_and_store_ngram_model(config=config)
-    breakpoint()
 
     if config.push_to_hub:
         push_model_to_hub(