diff --git a/train.py b/train.py index 2a0200fc2..83cf2439a 100644 --- a/train.py +++ b/train.py @@ -65,7 +65,7 @@ choices=["halving", "median", "none"], ) parser.add_argument("--n-startup-trials", help="Number of trials before using optuna sampler", type=int, default=10) - parser.add_argument("--n-evaluations", help="Number of evaluations for hyperparameter optimization", type=int, default=20) + parser.add_argument("--n-evaluations", help="Number of timesteps at which the environment is evaluated during for a single hyperparameter combination", type=int, default=20) parser.add_argument( "--storage", help="Database storage path if distributed optimization should be used", type=str, default=None )