From 3c7670e1b4f3641d8c3d6464c4a05b2fcf607d2f Mon Sep 17 00:00:00 2001 From: Robin Meersman Date: Mon, 6 Apr 2026 11:54:07 +0200 Subject: [PATCH] fix(train.py): replaced debugging iteration count with dynamically calculated count --- experiments/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/experiments/train.py b/experiments/train.py index 716a917..2e4bca7 100644 --- a/experiments/train.py +++ b/experiments/train.py @@ -19,8 +19,7 @@ if __name__ == "__main__": args.batch_size = args.num_envs * args.num_steps args.minibatch_size = args.batch_size // args.num_minibatches - # args.num_iterations = args.total_timesteps // args.batch_size - args.num_iterations = 5 + args.num_iterations = args.total_timesteps // args.batch_size run_name = f"{args.exp_name}__seed_{args.seed}__{int(time.time())}" env = make_env(args.config_path, args.num_envs)