Upload New File

d6ab59c8 · zhougaofeng · c6f1f140 · d6ab59c8
Commit d6ab59c8 authored Jun 11, 2024 by zhougaofeng
Hide whitespace changes
Inline Side-by-side

Showing with 14 additions and 0 deletions

src/train.py src/train.py +14 -0

No files found.
--- a/src/train.py
+++ b/src/train.py
+from llamafactory.train.tuner import run_exp
+
+
+def main():
+    run_exp()
+
+
+def _mp_fn(index):
+    # For xla_spawn (TPUs)
+    run_exp()
+
+
+if __name__ == "__main__":
+    main()