github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[NFC] polish colossalai/booster/plugin/low_level_zero_plugin.py code style (#4256) 

Co-authored-by: supercooledith <893754954@qq.com>
pull/4338/head
梁爽 2023-07-18 10:58:43 +08:00 committed by binmakeswell
parent c614a99d28
commit abe4f971e0
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
colossalai/booster/plugin/low_level_zero_plugin.py
View File

@ -208,10 +208,7 @@ class LowLevelZeroPlugin(DPPluginBase):
if optimizer is not None and \
not isinstance(optimizer, OptimizerWrapper):
optimizer = LowLevelZeroOptimizer(model.unwrap(),
optimizer,
self.zero_optim_config,
self.optim_kwargs,
optimizer = LowLevelZeroOptimizer(model.unwrap(), optimizer, self.zero_optim_config, self.optim_kwargs,
self.verbose)
return model, optimizer, criterion, dataloader, lr_scheduler