ray.train.DataConfig.set_train_total_resources

DataConfig.set_train_total_resources(num_train_cpus: float, num_train_gpus: float)

Set the total number of CPUs and GPUs used by training.

If CPU or GPU resource limits are not set, they will be set to the total cluster resources minus the resources used by training.