Commit 3d75ed5b authored by Unknown's avatar Unknown

changed to 20 test for high level

parent d5cda2c5
......@@ -172,7 +172,7 @@ if __name__ == "__main__":
if args.train:
high_level_policy_training(nb_steps=args.nb_steps, load_weights=args.load_weights, save_path=args.save_file,
tensorboard=args.tensorboard, nb_episodes_for_test=args.nb_episodes_for_test,
tensorboard=args.tensorboard, nb_episodes_for_test=20,
visualize=args.visualize)
if args.test:
......
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