updated econfigs

april-tools · Jul 31, 2024 · 7265aac · 7265aac
1 parent c23ff28
commit 7265aac
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 10 deletions.
diff --git a/econfigs/complex-squared-npcs.json b/econfigs/complex-squared-npcs.json
@@ -7,7 +7,7 @@
     "device": "cuda",
     "num-workers": 2,
     "patience-threshold": 1e-3,
-    "early-stop-patience": 30
+    "early-stop-patience": 25
   },
   "datasets": ["miniboone", "hepmass", "gas", "power"],
   "grid": {

diff --git a/econfigs/exp-sum-of-squares.json b/econfigs/exp-sum-of-squares.json
@@ -7,7 +7,7 @@
     "device": "cuda",
     "num-workers": 2,
     "patience-threshold": 1e-3,
-    "early-stop-patience": 30
+    "early-stop-patience": 25
   },
   "datasets": ["miniboone", "hepmass"],
   "grid": {
@@ -20,20 +20,38 @@
       "learning-rate": [1e-4, 5e-4, 1e-3, 5e-3]
     },
     "models": {
-      "hepmass|miniboone": {
+      "hepmass": {
         "ExpSOS": {
           "real": {
-            "num-units": 208,
+            "num-units": 128,
             "num-input-units": 128,
-            "mono-num-units": 8,
-            "mono-num-input-units": 8,
+            "mono-num-units": [2, 16],
+            "mono-num-input-units": [2, 16],
             "complex": false
           },
           "complex": {
-            "num-units": 128,
+            "num-units": 64,
             "num-input-units": 128,
-            "mono-num-units": 8,
-            "mono-num-input-units": 8,
+            "mono-num-units": [2, 16],
+            "mono-num-input-units": [2, 16],
+            "complex": true
+          }
+        }
+      },
+      "miniboone": {
+        "ExpSOS": {
+          "real": {
+            "num-units": 128,
+            "num-input-units": 64,
+            "mono-num-units": [2, 16],
+            "mono-num-input-units": [2, 16],
+            "complex": false
+          },
+          "complex": {
+            "num-units": 64,
+            "num-input-units": 64,
+            "mono-num-units": [2, 16],
+            "mono-num-input-units": [2, 16],
             "complex": true
           }
         }

diff --git a/econfigs/num-of-squares.json b/econfigs/num-of-squares.json
@@ -7,7 +7,7 @@
     "device": "cuda",
     "num-workers": 2,
     "patience-threshold": 1e-3,
-    "early-stop-patience": 30
+    "early-stop-patience": 25
   },
   "datasets": ["miniboone", "hepmass", "gas", "power"],
   "grid": {