add close

2025-07-09 21:56:27 +08:00 · 2025-07-09 21:56:27 +08:00 · 32077b02ed
parent b75aaaf130
commit 32077b02ed
1 changed files with 3 additions and 0 deletions
--- a/examples/arealite/gsm8k_sft.py
+++ b/examples/arealite/gsm8k_sft.py
@ -114,6 +114,9 @@ def main_sft():
            logger.commit(epoch, step, global_step, stats_tracker.export())
            global_step += 1

+    engine.destroy()
+    logger.close()
+

 if __name__ == "__main__":
    main_sft()