Browse Source

[NFC] Fix format for mixed precision (#4253)

* [NFC] polish colossalai/booster/mixed_precision/mixed_precision_base.py code style
pull/4338/head
Jianghai 1 year ago committed by binmakeswell
parent
commit
b366f1d99f
  1. 11
      colossalai/booster/mixed_precision/mixed_precision_base.py

11
colossalai/booster/mixed_precision/mixed_precision_base.py

@ -13,10 +13,11 @@ class MixedPrecision(ABC):
"""
@abstractmethod
def configure(self,
model: nn.Module,
optimizer: Optional[Optimizer] = None,
criterion: Optional[Callable] = None,
) -> Tuple[nn.Module, OptimizerWrapper, Callable]:
def configure(
self,
model: nn.Module,
optimizer: Optional[Optimizer] = None,
criterion: Optional[Callable] = None,
) -> Tuple[nn.Module, OptimizerWrapper, Callable]:
# TODO: implement this method
pass

Loading…
Cancel
Save