diff --git a/fam/llm/finetune.py b/fam/llm/finetune.py index eab55bb..2ae6d8b 100644 --- a/fam/llm/finetune.py +++ b/fam/llm/finetune.py @@ -373,6 +373,7 @@ def get_lr(it): properties={"finetune_jobid": finetune_jobid, "loss": round(lossf, 4)}, ) ) + break if __name__ == "__main__":