diff --git a/benchmark/tabular/model.py b/benchmark/tabular/model.py index c1decf7f7..f363ae972 100644 --- a/benchmark/tabular/model.py +++ b/benchmark/tabular/model.py @@ -404,12 +404,14 @@ class SDMKumoTabularSmallModel(SDMKumoTabularModel): ag_key = "SDM-KUMO-TABULAR-SMALL" ag_name = "SDMKumoTabularSmall" size = "small" + default_num_estimators = 8 class SDMKumoTabularMediumModel(SDMKumoTabularModel): ag_key = "SDM-KUMO-TABULAR-MEDIUM" ag_name = "SDMKumoTabularMedium" size = "medium" + default_num_estimators = 8 class SDMKumoTabularLargeModel(SDMKumoTabularModel): diff --git a/benchmark/tabular/scoringbench/models.py b/benchmark/tabular/scoringbench/models.py index a3b00be0d..98c932191 100644 --- a/benchmark/tabular/scoringbench/models.py +++ b/benchmark/tabular/scoringbench/models.py @@ -56,14 +56,14 @@ def _create_kumo_tabular( method="sdm_kumo_tabular_small", factory=partial(_create_kumo_tabular, size="small"), autocast_dtype=torch.float16, - num_estimators=16, + num_estimators=8, ), "kumo-tabular-medium": ModelConfig( name="KumoTabular-Medium", method="sdm_kumo_tabular_medium", factory=partial(_create_kumo_tabular, size="medium"), autocast_dtype=torch.float16, - num_estimators=16, + num_estimators=8, ), "kumo-tabular-large": ModelConfig( name="KumoTabular-Large", diff --git a/benchmark/tabular/talent/models.py b/benchmark/tabular/talent/models.py index 94ab858f7..35350a763 100644 --- a/benchmark/tabular/talent/models.py +++ b/benchmark/tabular/talent/models.py @@ -86,14 +86,14 @@ def _create_tabfm( "kumo-tabular-small": ModelConfig( name="KumoTabular-Small", factory=partial(_create_kumo_tabular, size="small"), - num_estimators=16, + num_estimators=8, autocast_dtype=torch.float16, max_classes=10, ), "kumo-tabular-medium": ModelConfig( name="KumoTabular-Medium", factory=partial(_create_kumo_tabular, size="medium"), - num_estimators=16, + num_estimators=8, autocast_dtype=torch.float16, max_classes=10, ),