update sample config

autogluon · Feb 12, 2025 · f348b12 · f348b12
1 parent 7896702
commit f348b12
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/sample_configs/resources/multimodal_frameworks.yaml b/sample_configs/resources/multimodal_frameworks.yaml
@@ -2,8 +2,10 @@ AutoGluon_branch:
   repo: https://github.com/autogluon/autogluon.git
   version: master
   params:  # MultimodalPredictor.fit(params)
-    presets: medium_quality
-    time_limit:  500
-    hyperparameters:
-      optimization.max_epochs: 50
-      optimization.learning_rate: 0.005
+    predictor_args:
+      presets: medium_quality
+    fit_args:
+      time_limit:  500
+      hyperparameters:
+        optimization.max_epochs: 50
+        optimization.learning_rate: 0.005