import fix

deepray-AI · Nov 17, 2023 · 43d032d · 43d032d
1 parent e7247a8
commit 43d032d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deepray/core/base_trainer.py b/deepray/core/base_trainer.py
@@ -258,7 +258,7 @@ def __init__(
 
     # TODO: fuhailin 
     # if isinstance(optimizer, optimizers.Optimizer):
-      self.optimizer = optimizer
+    self.optimizer = optimizer
     # else:
     #   raise ValueError("Not support opt.")
     self.use_float16 = common_flags.use_float16()

diff --git a/deepray/optimizers/__init__.py b/deepray/optimizers/__init__.py
@@ -49,4 +49,4 @@
 )
 from deepray.optimizers.yogi import Yogi
 from deepray.optimizers.cocob import COCOB
-from deepray.optimizers.adam import AdamOptimizer
+from deepray.optimizers.adam import Adam