From 34323fe17ac63fdaf63a2fb0bc226b5670e3e04d Mon Sep 17 00:00:00 2001 From: digger yu Date: Fri, 30 Jun 2023 08:35:46 +0800 Subject: [PATCH] fix ColossalaiOptimizer is not defined --- 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 ff8979d82401..db27ad0e8abe 100644 --- a/colossalai/engine/_base_engine.py +++ b/colossalai/engine/_base_engine.py @@ -12,7 +12,7 @@ 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