Skip to content
Snippets Groups Projects
Commit b0676728 authored by Gunnar Arctaedius's avatar Gunnar Arctaedius
Browse files

had to change config

parent 67806ac3
No related branches found
No related tags found
No related merge requests found
...@@ -182,9 +182,9 @@ def get_config(): ...@@ -182,9 +182,9 @@ def get_config():
parser.add_argument("--cuda_deterministic", action='store_false', parser.add_argument("--cuda_deterministic", action='store_false',
default=True, help="by default, make sure random seed effective. if set, bypass such function.") default=True, help="by default, make sure random seed effective. if set, bypass such function.")
parser.add_argument("--n_training_threads", type=int, parser.add_argument("--n_training_threads", type=int,
default=20, help="Number of torch threads for training") default=1, help="Number of torch threads for training")
parser.add_argument("--n_rollout_threads", type=int, parser.add_argument("--n_rollout_threads", type=int,
default=1, help="Number of parallel envs for training rollouts") default=20, help="Number of parallel envs for training rollouts")
parser.add_argument("--n_eval_rollout_threads", type=int, parser.add_argument("--n_eval_rollout_threads", type=int,
default=1, help="Number of parallel envs for evaluating rollouts") default=1, help="Number of parallel envs for evaluating rollouts")
parser.add_argument("--n_render_rollout_threads", type=int, parser.add_argument("--n_render_rollout_threads", type=int,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment