works

antipasto/train/train_adapter.py

@@ -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,

justfile

@@ -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