ColossalAI/colossalai/booster/mixed_precision/fp16_apex.py

6 lines
106 B
Python

from .mixed_precision_base import MixedPrecision
class FP16ApexMixedPrecision(MixedPrecision):
pass