Document missing runs in README and add train_missing.yaml

README.md

@@ -54,13 +54,25 @@ wandb_project=jaxgmg2_3phase_fast_tight
 
 Checkpoints are named `al_{alpha}_g_{discount_rate}_seed_{seed}_pa_1`.
 
+## Missing runs
+
+5 runs from the initial sweep failed and are not present:
+
+- `al_0.0_g_0.97_seed_100_pa_1`
+- `al_0.0_g_0.99_seed_100_pa_1`
+- `al_0.45_g_0.97_seed_103_pa_1`
+- `al_0.45_g_0.98_seed_100_pa_1`
+- `al_0.4_g_0.99_seed_104_pa_1`
+
 ## Reproduced with
 
-See `train.yaml` (initial sweep) and `train_extended.yaml` (extended sweep) in this repository. Run with:
+See `train.yaml` (initial sweep), `train_extended.yaml` (extended sweep), and
+`train_missing.yaml` (5 failed runs from initial sweep) in this repository. Run with:
 
 ```bash
 timaeus run train.yaml
 timaeus run train_extended.yaml
+timaeus run train_missing.yaml
 ```
 
 from the [timaeus monorepo](https://github.com/timaeus-research/timaeus).

train_missing.yaml

@@ -0,0 +1,41 @@
+parameters:
+  project_name: jaxgmg2_3phase_fast_tight
+  action: rl
+  rl_action: train
+
+  lr: 5e-5
+  cheese_loc: any
+  env_layout: open
+  mask_type: first_episode
+  use_prev_action: false
+
+  num_total_env_steps: 10_000_000_000
+  num_levels: 9600
+  grad_acc_per_chunk: 4
+  num_rollout_steps: 64
+
+  ckpt_dir: jaxgmg2_3phase_fast_tight
+  eval_schedule: "0:1,250:2,500:5,2000:10"
+  f_str_ckpt: "al_{alpha}_g_{discount_rate}_seed_{seed}_pa_1"
+
+  use_wandb: true
+  use_hf: true
+  wandb_project: jaxgmg2_3phase_fast_tight
+  ntfy: david_jaxgmg
+
+sweep:
+- - alpha: 0.0
+    discount_rate: 0.97
+    seed: 100
+  - alpha: 0.0
+    discount_rate: 0.99
+    seed: 100
+  - alpha: 0.45
+    discount_rate: 0.97
+    seed: 103
+  - alpha: 0.45
+    discount_rate: 0.98
+    seed: 100
+  - alpha: 0.4
+    discount_rate: 0.99
+    seed: 104