support DeepSpeed.

2026-10-17 00:23:35 +00:00 · 2024-12-24 10:59:21 +05:30
parent cf9be17819
commit 6329e9ed07
4 changed files with 140 additions and 52 deletions
@@ -33,7 +33,7 @@ def main():
        trainer.prepare_for_training()
        trainer.prepare_trackers()
        trainer.train()
-        trainer.evaluate()
+        # trainer.evaluate()

    except KeyboardInterrupt:
        logger.info("Received keyboard interrupt. Exiting...")