From 8689e8da65ed02a13975956d70216d7f33db1688 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 19 Jun 2022 20:58:36 +0200
Subject: Fix lr and optim conf

---
 training/run.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'training/run.py')

diff --git a/training/run.py b/training/run.py
index 0e17b4e..68cedc7 100644
--- a/training/run.py
+++ b/training/run.py
@@ -41,8 +41,8 @@ def run(config: DictConfig) -> Optional[float]:
         network=network,
         mapping=datamodule.mapping,
         loss_fn=loss_fn,
-        optimizer_configs=config.optimizers,
-        lr_scheduler_configs=config.lr_schedulers,
+        optimizer_config=config.optimizer,
+        lr_scheduler_config=config.lr_scheduler,
         _recursive_=False,
     )
 
-- 
cgit v1.2.3-70-g09d2