updated num-of-squares.json econfig

april-tools · Jul 24, 2024 · 743b851 · 743b851
1 parent fa2d2f2
commit 743b851
Showing 1 changed file with 14 additions and 21 deletions.
diff --git a/econfigs/num-of-squares.json b/econfigs/num-of-squares.json
@@ -1,69 +1,62 @@
 {
   "common": {
-    "tboard-path": "tboard-runs/uci-data",
-    "checkpoint-path": "checkpoints/uci-data",
+    "tboard-path": "tboard-runs/uci-data-num-squares",
+    "checkpoint-path": "checkpoints/uci-data-num-squares",
     "save-checkpoint": false,
-    "num-epochs": 1200,
+    "num-epochs": 1000,
     "device": "cuda",
     "num-workers": 2,
     "patience-threshold": 1e-3,
     "early-stop-patience": 30
   },
-  "datasets": ["power", "gas", "hepmass", "miniboone"],
+  "datasets": ["miniboone", "hepmass", "gas", "power"],
   "grid": {
     "common": {
       "region-graph": "rnd-bt",
       "region-graph-sd": true,
       "batch-size": 512,
       "optimizer": "Adam",
-      "verbose": true
+      "verbose": true,
+      "learning-rate": [1e-4, 5e-4, 1e-3, 5e-3]
     },
     "models": {
       "power": {
         "MPC": {
-          "learning-rate": 1e-3,
-          "num-components,num-units,num-input-units": [[1, 256, 512], [4, 178, 128], [16, 106, 32], [64, 58, 8]]
+          "num-components,num-units,num-input-units": [[4, 160, 64], [8, 120, 32], [16, 90, 16]]
         },
         "SOS": {
           "real": {
-            "learning-rate": 1e-3,
-            "num-components,num-units,num-input-units": [[1, 256, 512], [4, 178, 128], [16, 106, 32], [64, 58, 8]]
+            "num-components,num-units,num-input-units": [[4, 160, 64], [8, 120, 32], [16, 90, 16]]
           }
         }
       },
       "gas": {
         "MPC": {
-          "learning-rate": 1e-3,
-          "num-components,num-units,num-input-units": [[1, 256, 512], [4, 174, 128], [16, 104, 32], [64, 56, 8]]
+          "num-components,num-units,num-input-units": [[4, 158, 64], [8, 118, 32], [16, 88, 16]]
         },
         "SOS": {
           "real": {
-            "learning-rate": 1e-3,
-            "num-components,num-units,num-input-units": [[1, 256, 512], [4, 174, 128], [16, 104, 32], [64, 56, 8]]
+            "num-components,num-units,num-input-units": [[4, 158, 64], [8, 118, 32], [16, 88, 16]]
           }
         }
       },
       "hepmass": {
         "MPC": {
-          "learning-rate": 1e-3,
-          "num-components,num-units,num-input-units": [[1, 128, 512], [4, 94, 128], [16, 59, 32], [64, 33, 8]]
+          "num-components,num-units,num-input-units": [[4, 78, 32], [8, 58, 16], [16, 42, 8]]
         },
         "SOS": {
           "real": {
-            "learning-rate": 1e-3,
-            "num-components,num-units,num-input-units": [[1, 128, 512], [4, 94, 128], [16, 59, 32], [64, 33, 8]]
+            "num-components,num-units,num-input-units": [[4, 78, 32], [8, 58, 16], [16, 42, 8]]
           }
         }
       },
       "miniboone": {
         "MPC": {
-          "learning-rate": 1e-3,
-          "num-components,num-units,num-input-units": [[1, 64, 512], [4, 52, 128], [16, 35, 32], [64, 21, 8]]
+          "num-components,num-units,num-input-units": [[4, 72, 16], [8, 52, 8], [16, 38, 4]]
         },
         "SOS": {
           "real": {
-            "learning-rate": 1e-3,
-            "num-components,num-units,num-input-units": [[1, 64, 512], [4, 52, 128], [16, 35, 32], [64, 21, 8]]
+            "num-components,num-units,num-input-units": [[4, 72, 16], [8, 52, 8], [16, 38, 4]]
           }
         }
       }