refine training

2025-05-12 14:19:00 +08:00
parent f17558a4c4
commit 91fbb24e17
3 changed files with 216 additions and 26 deletions
--- a/run_single.sh
+++ b/run_single.sh
@@ -0,0 +1,4 @@
+accelerate launch \
+    train.py \
+    --output_path models/nexus_v3 \
+    --steps_per_epoch 4000