1
Fork 0

fix(train.py): replaced debugging iteration count with dynamically calculated count

This commit is contained in:
Robin Meersman 2026-04-06 11:54:07 +02:00
parent ca17574272
commit 3c7670e1b4

View file

@ -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)