diff --git a/baselines/fedavgm/fedavgm/conf/model/cnn.yaml b/baselines/fedavgm/fedavgm/conf/model/cnn.yaml
index 29bbd2aba540..c25463693c7f 100644
--- a/baselines/fedavgm/fedavgm/conf/model/cnn.yaml
+++ b/baselines/fedavgm/fedavgm/conf/model/cnn.yaml
@@ -2,4 +2,4 @@
 _target_: fedavgm.models.cnn
 input_shape: ${dataset.input_shape}
 num_classes: ${dataset.num_classes}
-lr: ${client.lr}
\ No newline at end of file
+learning_rate: ${client.lr}
\ No newline at end of file
diff --git a/baselines/fedavgm/fedavgm/conf/model/tf_example.yaml b/baselines/fedavgm/fedavgm/conf/model/tf_example.yaml
index ff36bffb3736..8c2a670ee978 100644
--- a/baselines/fedavgm/fedavgm/conf/model/tf_example.yaml
+++ b/baselines/fedavgm/fedavgm/conf/model/tf_example.yaml
@@ -2,4 +2,4 @@
 _target_: fedavgm.models.tf_example
 input_shape: ${dataset.input_shape}
 num_classes: ${dataset.num_classes}
-lr: ${client.lr}
\ No newline at end of file
+learning_rate: ${client.lr}
\ No newline at end of file