fix(train.py): replaced debugging iteration count with dynamically calculated count
This commit is contained in:
parent
ca17574272
commit
3c7670e1b4
1 changed files with 1 additions and 2 deletions
|
|
@ -19,8 +19,7 @@ if __name__ == "__main__":
|
||||||
|
|
||||||
args.batch_size = args.num_envs * args.num_steps
|
args.batch_size = args.num_envs * args.num_steps
|
||||||
args.minibatch_size = args.batch_size // args.num_minibatches
|
args.minibatch_size = args.batch_size // args.num_minibatches
|
||||||
# args.num_iterations = args.total_timesteps // args.batch_size
|
args.num_iterations = args.total_timesteps // args.batch_size
|
||||||
args.num_iterations = 5
|
|
||||||
run_name = f"{args.exp_name}__seed_{args.seed}__{int(time.time())}"
|
run_name = f"{args.exp_name}__seed_{args.seed}__{int(time.time())}"
|
||||||
env = make_env(args.config_path, args.num_envs)
|
env = make_env(args.config_path, args.num_envs)
|
||||||
|
|
||||||
|
|
|
||||||
Reference in a new issue