fix

GreenWizard2015 · Apr 10, 2024 · bf4ed1e · bf4ed1e
1 parent 0531502
commit bf4ed1e
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/configs/experiments/decoder/mlp-huge.json b/configs/experiments/decoder/mlp-huge.json
@@ -1,11 +1,15 @@
 {
-  "name": "mlp",
-  "residual": true,
-  "mlp blocks": 5,
-  "mlp": {
-    "shared": false,
-    "sizes": [768, 768, 768],
-    "activation": "relu",
-    "final activation": "linear"
+  "model": {
+    "decoder": {
+      "name": "mlp",
+      "residual": true,
+      "mlp blocks": 5,
+      "mlp": {
+        "shared": false,
+        "sizes": [768, 768, 768],
+        "activation": "relu",
+        "final activation": "linear"
+      }
+    }
   }
 }