From bf33b020c07b3db48ce8332827eef3805fa26dae Mon Sep 17 00:00:00 2001 From: Sven <6873303+gieses@users.noreply.github.com> Date: Sun, 15 Oct 2023 21:31:07 +0200 Subject: [PATCH] update lr to learning_rate --- xirt/xirtnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/xirt/xirtnet.py b/xirt/xirtnet.py index 489e313..0556a30 100644 --- a/xirt/xirtnet.py +++ b/xirt/xirtnet.py @@ -407,7 +407,7 @@ def compile(self): """ # set optimizer if self.learning_p["optimizer"] == "adam": - opt = optimizers.Adam(lr=self.learning_p["learningrate"]) + opt = optimizers.Adam(learning_rate=self.learning_p["learningrate"]) elif self.learning_p["optimizer"].lower() == "sgd": opt = optimizers.SGD(learning_rate=self.learning_p["learningrate"],