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
}