Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "f27d801a13c2feed37ea676e38c6ef8831d58e76"
Unverified Commit 09fe9dc7 authored by digger yu's avatar digger yu Committed by GitHub
Browse files

[nfc]fix ColossalaiOptimizer is not defined (#4122)

parent edd75a59
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/engine/_base_engine.py
View file @ 09fe9dc7
...@@ -12,7 +12,7 @@ from colossalai.engine.gradient_handler import BaseGradientHandler ...@@ -12,7 +12,7 @@ from colossalai.engine.gradient_handler import BaseGradientHandler
from colossalai.engine.schedule import BaseSchedule, InterleavedPipelineSchedule, NonPipelineSchedule, PipelineSchedule from colossalai.engine.schedule import BaseSchedule, InterleavedPipelineSchedule, NonPipelineSchedule, PipelineSchedule
from colossalai.logging import get_dist_logger from colossalai.logging import get_dist_logger
from colossalai.zero.legacy.gemini import BaseOpHook, register_ophooks_recursively from colossalai.zero.legacy.gemini import BaseOpHook, register_ophooks_recursively
from colossalai.nn.optimizer import ColossalaiOptimizer
class Engine: class Engine:
"""Basic engine class for training and evaluation. It runs a specific process method """Basic engine class for training and evaluation. It runs a specific process method
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment