Closed KYLN24 closed 4 months ago
super().__init__()
DeepSpeedEngine.__init__()
PipelineEngine
pipeline_engine.py
逻辑上有些问题。ColliePipelineEngine的init大多数与PipelineEngine相同,所以之前调用的是DeepSpeedEngine的init。
为了动态匹配ds的更新,可以改成直接用PipelineEngine的init方法,之后注册一些collie需要用到的属性。
super().__init__()
instead ofDeepSpeedEngine.__init__()
to correctly initialize the super classPipelineEngine
.pipeline_engine.py
by Black formatter.