How to tell RLLIB tune to run that many number of episodes

Hey, I am using RLLIB PPO with a custom environment. I am using ray.tune to tune hyperparameters. How do I tell ray.tune to run let’s say 200 episodes.

In the configuration you pass you ray.tune, you can specify stopping condition:

config = {
    'stop': {
        'episodes_total': 200,
    },
}
ray.tune(**config)
1 Like