From 4f3105a7996653056bc38c42d27a4b58496601a9 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Tue, 11 Apr 2023 12:08:30 +0200
Subject: [PATCH] Renamed optimizer_params to optimizer_configs
---
src/ptbench/engine/callbacks.py | 2 +-
src/ptbench/models/densenet.py | 2 +-
src/ptbench/models/pasa.py | 4 +++-
3 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/src/ptbench/engine/callbacks.py b/src/ptbench/engine/callbacks.py
index a01a7e19..0080676a 100644
--- a/src/ptbench/engine/callbacks.py
+++ b/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
diff --git a/src/ptbench/models/densenet.py b/src/ptbench/models/densenet.py
index 17373b79..ab7a1f71 100644
--- a/src/ptbench/models/densenet.py
+++ b/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,
):
diff --git a/src/ptbench/models/pasa.py b/src/ptbench/models/pasa.py
index b31fa21d..d5657218 100644
--- a/src/ptbench/models/pasa.py
+++ b/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()
--
GitLab