Fix learning rate parameter in SSLContainer

CHANGELOG.md

@@ -62,6 +62,8 @@ mounting and running matplotblib on some machines. Re-instantiated a disabled te
 - ([#509](https://github.com/microsoft/InnerEye-DeepLearning/pull/509)) Fix issue where model checkpoints were not loaded
 in inference-only runs when using lightning containers.
 - ([#553](https://github.com/microsoft/InnerEye-DeepLearning/pull/553)) Fix incomplete test data module setup in Lightning inference.
+- ([#557](https://github.com/microsoft/InnerEye-DeepLearning/pull/557)) Fix issue where learning rate was not set
+  correctly in the SimCLR module
 
 ### Removed
 

InnerEye/ML/SSL/lightning_containers/ssl_container.py

@@ -149,7 +149,7 @@ def create_model(self) -> LightningModule:
                                                     gpus=self.total_num_gpus,
                                                     num_samples=self.data_module.num_samples,
                                                     batch_size=self.data_module.batch_size,
-                                                    lr=self.l_rate,
+                                                    learning_rate=self.l_rate,
                                                     max_epochs=self.num_epochs)
         elif self.ssl_training_type == SSLTrainingType.BYOL:
             model = BYOLInnerEye(encoder_name=self.ssl_encoder.value,