fix: change eval interval

This commit is contained in:
Ting-Jun Wang 2024-05-24 15:17:43 +08:00
parent d0a3d80b77
commit f2cf49a05e
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354

View File

@ -31,8 +31,8 @@ def parse_args():
parser.add_argument('--buffer_size', type=int, default=int(1e5)) parser.add_argument('--buffer_size', type=int, default=int(1e5))
parser.add_argument('--target_update_interval', type=int, default=10000) parser.add_argument('--target_update_interval', type=int, default=10000)
# training hyperparameters # training hyperparameters
parser.add_argument('--max_steps', type=int, default=int(2e5)) parser.add_argument('--max_steps', type=int, default=int(1e5))
parser.add_argument('--eval_interval', type=int, default=5000) parser.add_argument('--eval_interval', type=int, default=2000)
# others # others
parser.add_argument('--save_root', type=Path, default='./submissions') parser.add_argument('--save_root', type=Path, default='./submissions')
parser.add_argument("--exp-name", type=str, default=os.path.basename(__file__).rstrip(".py"), parser.add_argument("--exp-name", type=str, default=os.path.basename(__file__).rstrip(".py"),