From f40efc451e942838b9c055b5e586d9b329f702e7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pawe=C5=82=20Czy=C5=BC?= Date: Fri, 12 Apr 2024 17:47:39 +0200 Subject: [PATCH] Use default model architectures. --- workflows/benchmark/v2/config.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/workflows/benchmark/v2/config.py b/workflows/benchmark/v2/config.py index 54772004..a1cdb439 100644 --- 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(), }