Jae Young Lee · Jae Young Lee
--- a/high_level_policy_main.py

+ 1

− 1
+++ b/high_level_policy_main.py

+ 1

− 1
 @@ -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: