HPUPrecisionPlugin¶
- class lightning.pytorch.plugins.precision.HPUPrecisionPlugin(precision, opt_level='O2', bf16_file_path=None, fp32_file_path=None, verbose=False)[source]¶
Bases:
lightning.pytorch.plugins.precision.precision_plugin.PrecisionPlugin
Plugin that enables bfloat/half support on HPUs.
Warning
This is an experimental feature.