o wi@s(ddlZddlmZGdddeZdS)N)MixedPrecisionPlugincs,eZdZd dededdffdd ZZS) NeMoMixedPrecisionPlugin init_scalegrowth_intervalreturnNcs&tjddtjjj||d|_dS)N) precision)rr)super__init__torchcudaamp GradScalerscaler)selfrr __class__h/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/collections/common/parts/ptl_overrides.pyr sz!NeMoMixedPrecisionPlugin.__init__)rr)__name__ __module__ __qualname__floatintr __classcell__rrrrrs$r)r #lightning.pytorch.plugins.precisionrrrrrrs