"train_batch_size": "auto", "train_micro_batch_size_per_gpu": "auto", "wall_clock_breakdown": False, } # Init Ray cluster ray.init(address="auto") print(f" Ray CLuster resources:\n {ray.cluster_resources()}") # Prepare Ray dataset and batch mapper dataset = prepare_dataset(args....