diff --git a/configs/madpo_iter_che_config.yaml b/configs/madpo_iter_che_config.yaml
index 04aad19..d7e3243 100644
--- a/configs/madpo_iter_che_config.yaml
+++ b/configs/madpo_iter_che_config.yaml
@@ -45,6 +45,8 @@ madpo_iter:
   num_generations: 4
   preference_num_candidates: 20
   comparator_policy: current
+  comparator_generation_mode: decentralized
+  comparator_centralized_agent_index: 0
   comparator_model_name: null
   comparator_agents: null
   comparator_devices: null
diff --git a/configs/marlhf_iter_che_config.yaml b/configs/marlhf_iter_che_config.yaml
index da69184..163ec3a 100644
--- a/configs/marlhf_iter_che_config.yaml
+++ b/configs/marlhf_iter_che_config.yaml
@@ -46,6 +46,8 @@ marlhf_iter:
   num_generations: 4
   preference_num_candidates: 20
   comparator_policy: current
+  comparator_generation_mode: decentralized
+  comparator_centralized_agent_index: 0
   comparator_model_name: null
   comparator_agents: null
   comparator_devices: null