From 591bbd8d892eba820ed28662c9d72d2427f2789c Mon Sep 17 00:00:00 2001 From: Uncertainty Baselines Team Date: Thu, 29 Sep 2022 01:08:47 -0700 Subject: [PATCH] Implements reweighted training with ensemble. Supports specifying the number of splits. PiperOrigin-RevId: 477657105 --- .../shoshin/configs/waterbirds_resnet_config_reweighting.py | 1 + 1 file changed, 1 insertion(+) diff --git a/experimental/shoshin/configs/waterbirds_resnet_config_reweighting.py b/experimental/shoshin/configs/waterbirds_resnet_config_reweighting.py index af8bbe12b..38098c30c 100644 --- a/experimental/shoshin/configs/waterbirds_resnet_config_reweighting.py +++ b/experimental/shoshin/configs/waterbirds_resnet_config_reweighting.py @@ -32,6 +32,7 @@ def get_config() -> ml_collections.ConfigDict: model.dropout_rate = 0.2 config.train_bias = False + config.train_weight_ensemble = True reweighting = config.reweighting reweighting.do_reweighting = True