MixedPrecision

class tensorplay.distributed.fsdp.MixedPrecision(param_dtype: Any = None, reduce_dtype: Any = None, buffer_dtype: Any = None, keep_low_precision_grads: bool = False, cast_forward_inputs: bool = False, cast_root_forward_inputs: bool = True, _module_classes_to_ignore: collections.abc.Sequence[type] = (<class 'tensorplay.nn.modules.batchnorm._BatchNorm'>, ))[source]