From 43d032d05d815db771ae373c12d0797c841dcf9e Mon Sep 17 00:00:00 2001 From: Vincent Date: Fri, 17 Nov 2023 22:21:36 +0800 Subject: [PATCH] import fix --- deepray/core/base_trainer.py | 2 +- deepray/optimizers/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/deepray/core/base_trainer.py b/deepray/core/base_trainer.py index bd2f75d..aca2bcf 100644 --- 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 index f9b81f1..8eaff46 100644 --- 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