Skip to content

Y
ygo-agent
Commit 4d07e48e authored by sbl1996@126.com's avatar sbl1996@126.com
Browse files
Options

eval create replay dir

parent 01510089
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
scripts/eval.py
View file @ 4d07e48e
...@@ -98,6 +98,8 @@ if __name__ == "__main__": ...@@ -98,6 +98,8 @@ if __name__ == "__main__":
if args.record: if args.record:
assert args.num_envs == 1, "Recording only works with a single environment" assert args.num_envs == 1, "Recording only works with a single environment"
assert args.verbose, "Recording only works with verbose mode" assert args.verbose, "Recording only works with verbose mode"
if not os.path.exists("replay"):
os.makedirs("replay")
args.env_threads = min(args.env_threads or args.num_envs, args.num_envs) args.env_threads = min(args.env_threads or args.num_envs, args.num_envs)
args.torch_threads = args.torch_threads or int(os.getenv("OMP_NUM_THREADS", "4")) args.torch_threads = args.torch_threads or int(os.getenv("OMP_NUM_THREADS", "4"))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment