Skip to content

Commit 82348b3

Browse files
committed
fix: correct optimizer reference in Sup3rGan class for consistency
1 parent cda5917 commit 82348b3

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

sup3r/models/abstract.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,7 @@ def update_optimizer_gen(self, **kwargs):
356356
conf.update(**kwargs)
357357
self._optimizer_config = conf
358358
if self._optimizer is not None:
359-
self._optimizer = self.optimizer.__class__.from_config(conf)
359+
self._optimizer = self._optimizer.__class__.from_config(conf)
360360

361361
@property
362362
def history(self):

sup3r/models/base.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -394,8 +394,8 @@ def update_optimizer_disc(self, **kwargs):
394394
conf = self._optimizer_disc_config.copy()
395395
conf.update(**kwargs)
396396
self._optimizer_disc_config = conf
397-
if self.optimizer_disc is not None:
398-
self._optimizer_disc = self.optimizer_disc.__class__.from_config(
397+
if self._optimizer_disc is not None:
398+
self._optimizer_disc = self._optimizer_disc.__class__.from_config(
399399
conf
400400
)
401401

@@ -677,7 +677,7 @@ def train(self, batch_handler, config=None, **kwargs):
677677
self._optimizer = None
678678
self._optimizer_disc = None
679679

680-
if self.optimizer is None or self.optimizer_disc is None:
680+
if self._optimizer is None or self._optimizer_disc is None:
681681
with self._training_scope():
682682
self._optimizer = self.init_optimizer(
683683
self._optimizer_config, learning_rate=None

0 commit comments

Comments
 (0)