works

2026-09-12 12:04:50 +08:00 · 2026-01-13 09:10:28 +08:00
parent c1f0f0e55e
commit be859ad438
2 changed files with 9 additions and 2 deletions
@@ -892,7 +892,6 @@ def train_epoch(
    loss_layers,
    loss_layer_indices,
    opt,
-    aggregator,
    scheduler,
    config: TrainingConfig,
    epoch: int,
@@ -1863,7 +1862,6 @@ def train_model(config: TrainingConfig):
            loss_layers=loss_layers,
            loss_layer_indices=loss_layer_indices,
            opt=opt,
-            aggregator=aggregator,
            scheduler=scheduler,
            config=config,
            epoch=epoch,
@@ -0,0 +1,9 @@
+
+
+default:
+    uv run python nbs/train.py tiny --quick
+    uv run python nbs/train.py tiny
+    uv run python nbs/train.py q06b-24gb
+    # uv run nbs/test_reload.py
+    uv run python nbs/train.py gemma1b-24gb
+    uv run python nbs/train.py q4b-24gb