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

Co-authored-by: supercooledith <893754954@qq.com>
pull/4338/head
梁爽 1 year ago committed by binmakeswell
parent c614a99d28
commit abe4f971e0

@ -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

Loading…
Cancel
Save