Renamed optimizer_params to optimizer_configs

src/ptbench/engine/callbacks.py

@@ -46,7 +46,7 @@ class LoggingCallback(Callback):
         self.log("total_time", current_time)
         self.log("eta", eta_seconds)
         self.log("loss", numpy.average(self.training_loss))
-        self.log("learning_rate", pl_module.hparams["optimizer_params"]["lr"])
+        self.log("learning_rate", pl_module.hparams["optimizer_configs"]["lr"])
         self.log("validation_loss", numpy.average(self.validation_loss))
 
         queue_retries = 0

src/ptbench/models/densenet.py

@@ -21,7 +21,7 @@ class Densenet(pl.LightningModule):
         criterion,
         criterion_valid,
         optimizer,
-        optimizer_params,
+        optimizer_configs,
         pretrained=False,
         nb_channels=3,
     ):

src/ptbench/models/pasa.py

@@ -33,7 +33,9 @@ class PASA(pl.LightningModule):
     Based on paper by [PASA-2019]_.
     """
 
-    def __init__(self, criterion, criterion_valid, optimizer, optimizer_params):
+    def __init__(
+        self, criterion, criterion_valid, optimizer, optimizer_configs
+    ):
         super().__init__()
 
         self.save_hyperparameters()