From 09fe9dc704dd388f08e3b19dca65d3d0be64f106 Mon Sep 17 00:00:00 2001
From: digger yu <digger-yu@outlook.com>
Date: Fri, 30 Jun 2023 17:23:22 +0800
Subject: [PATCH] [nfc]fix ColossalaiOptimizer is not defined (#4122)
---
colossalai/engine/_base_engine.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/colossalai/engine/_base_engine.py b/colossalai/engine/_base_engine.py
index ff8979d82..db27ad0e8 100644
--- a/colossalai/engine/_base_engine.py
+++ b/colossalai/engine/_base_engine.py
@@ -12,7 +12,7 @@ from colossalai.engine.gradient_handler import BaseGradientHandler
from colossalai.engine.schedule import BaseSchedule, InterleavedPipelineSchedule, NonPipelineSchedule, PipelineSchedule
from colossalai.logging import get_dist_logger
from colossalai.zero.legacy.gemini import BaseOpHook, register_ophooks_recursively
-
+from colossalai.nn.optimizer import ColossalaiOptimizer
class Engine:
"""Basic engine class for training and evaluation. It runs a specific process method