From d273b0221ca2a7ef0ac89c2a7ae9e59129c39d92 Mon Sep 17 00:00:00 2001 From: Pranjal Tandon Date: Fri, 5 Apr 2019 01:12:31 +0530 Subject: [PATCH] Update main.py --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 69affa3..aca7b12 100644 --- a/main.py +++ b/main.py @@ -28,7 +28,7 @@ parser.add_argument('--automatic_entropy_tuning', type=bool, default=False, meta help='Temperature parameter α automaically adjusted.') parser.add_argument('--seed', type=int, default=456, metavar='N', help='random seed (default: 456)') -parser.add_argument('--batch_size', type=int, default=100, metavar='N', +parser.add_argument('--batch_size', type=int, default=256, metavar='N', help='batch size (default: 256)') parser.add_argument('--num_steps', type=int, default=1000001, metavar='N', help='maximum number of steps (default: 1000000)')