update lr to learning_rate

Rappsilber-Laboratory · Oct 15, 2023 · bf33b02 · bf33b02
1 parent 837fde4
commit bf33b02
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"],