- ray.train.torch.accelerate(amp: bool = False) None [source]#
Enables training optimizations.
amp – If true, perform training with automatic mixed precision. Otherwise, use full precision.
train.torch.acceleratecannot be called more than once, and it must be called before any other
PublicAPI (beta): This API is in beta and may change before becoming stable.