Checkpoint 5 - incomplete

training/bc_pretrain.py

@@ -15,7 +15,7 @@ Usage::
 
     python -m training.bc_pretrain \\
         --demos training/demos.npz \\
-        --out training/runs/bc_flock
+        --out training/runs/bc
 """
 
 from __future__ import annotations
@@ -83,7 +83,7 @@ def policy_forward_mean(policy, obs_batch):
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--demos", default="training/demos.npz")
-    parser.add_argument("--out", default="training/runs/bc_solo")
+    parser.add_argument("--out", default="training/runs/bc")
     parser.add_argument("--epochs", type=int, default=60)
     parser.add_argument("--batch-size", type=int, default=256)
     parser.add_argument("--lr", type=float, default=1e-3)