Use default model architectures.

cbg-ethz · Apr 12, 2024 · f40efc4 · f40efc4
1 parent 94115b5
commit f40efc4
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/workflows/benchmark/v2/config.py b/workflows/benchmark/v2/config.py
@@ -19,12 +19,10 @@
 # Note that each estimator implements `IMutualInformationPointEstimator` interface
 
 ESTIMATORS_DICT = {
-    "NWJ-10_5": estimators.NWJEstimator(verbose=False, hidden_layers=(10, 5)),
-    "MINE-10_5": estimators.MINEEstimator(verbose=False, hidden_layers=(10, 5)),
-    "InfoNCE-10_5": estimators.InfoNCEEstimator(verbose=False, hidden_layers=(10, 5)),
-    "Donsker-Varadhan-10_5": estimators.DonskerVaradhanEstimator(
-        verbose=False, hidden_layers=(10, 5)
-    ),
+    "NWJ": estimators.NWJEstimator(verbose=False),
+    "MINE": estimators.MINEEstimator(verbose=False),
+    "InfoNCE": estimators.InfoNCEEstimator(verbose=False),
+    "Donsker-Varadhan": estimators.DonskerVaradhanEstimator(verbose=False),
     "KSG-10": estimators.KSGEnsembleFirstEstimator(neighborhoods=(10,)),
     "CCA": estimators.CCAMutualInformationEstimator(),
 }