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
}