github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-22 11:13:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: supercooledith <893754954@qq.com>
This commit is contained in:
梁爽 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 \ if optimizer is not None and \
not isinstance(optimizer, OptimizerWrapper): not isinstance(optimizer, OptimizerWrapper):
optimizer = LowLevelZeroOptimizer(model.unwrap(), optimizer = LowLevelZeroOptimizer(model.unwrap(), optimizer, self.zero_optim_config, self.optim_kwargs,
optimizer,
self.zero_optim_config,
self.optim_kwargs,
self.verbose) self.verbose)
return model, optimizer, criterion, dataloader, lr_scheduler return model, optimizer, criterion, dataloader, lr_scheduler