Add DB and FL runs to sanity checks.

alexhernandezgarcia · Dec 8, 2023 · d3b0126 · d3b0126
1 parent 08351bb
commit d3b0126
Showing 1 changed file with 48 additions and 0 deletions.
diff --git a/mila/dev/sanity_check_runs.yaml b/mila/dev/sanity_check_runs.yaml
@@ -24,6 +24,20 @@ jobs:
         __value__: grid
         length: 10
       gflownet: trajectorybalance
+  - slurm:
+      job_name: sanity-grid-db
+    script:
+      env:
+        __value__: grid
+        length: 10
+      gflownet: detailedbalance
+  - slurm:
+      job_name: sanity-grid-fl
+    script:
+      env:
+        __value__: grid
+        length: 10
+      gflownet: forwardlooking
   # Tetris
   - slurm:
       job_name: sanity-tetris-fm
@@ -43,6 +57,40 @@ jobs:
         height: 10
       gflownet: trajectorybalance
       proxy: tetris
+  # Mini-Tetris
+  - slurm:
+      job_name: sanity-mintetris-fm
+    script:
+      env:
+        __value__: tetris
+        width: 3
+        height: 10
+        pieces: ["J", "L", "S", "Z"]
+        allow_eos_before_full: True
+      gflownet: flowmatch
+      proxy: tetris
+  - slurm:
+      job_name: sanity-mintetris-tb
+    script:
+      env:
+        __value__: tetris
+        width: 3
+        height: 10
+        pieces: ["J", "L", "S", "Z"]
+        allow_eos_before_full: True
+      gflownet: trajectorybalance
+      proxy: tetris
+  - slurm:
+      job_name: sanity-mintetris-fl
+    script:
+      env:
+        __value__: tetris
+        width: 3
+        height: 10
+        pieces: ["J", "L", "S", "Z"]
+        allow_eos_before_full: True
+      gflownet: forwardlooking
+      proxy: tetris
   # Ctorus
   - slurm:
       job_name: sanity-ctorus