Minimal training script for testing train dashboard

minimal_train.py

import ray 
from ray.train.torch import TorchTrainer
from ray.train import RunConfig, ScalingConfig
import time

def train_func():
    print("Training Starts")
    time.sleep(100)

datasets = {
    "train": ray.data.range(100),
    "val": ray.data.range(100)
}

trainer = TorchTrainer(
    train_func,
    run_config=RunConfig(
        name="my_train_run",
        storage_path="/mnt/cluster_storage"
    ),
    scaling_config=ScalingConfig(
        num_workers=8,
        use_gpu=True
    ),
    datasets=datasets
)
trainer.fit()