FP16OptimizerApex.ConfigΒΆ

Component: FP16OptimizerApex

class FP16OptimizerApex.Config[source]

Bases: FP16Optimizer.Config

All Attributes (including base classes)

opt_level: str = 'O2'
init_loss_scale: Optional[int] = None
min_loss_scale: Optional[float] = None

Default JSON

{
    "opt_level": "O2",
    "init_loss_scale": null,
    "min_loss_scale": null
}