diff --git a/main.py b/main.py index 1253cd1..f280892 100644 --- a/main.py +++ b/main.py @@ -30,7 +30,7 @@ parser.add_argument('--seed', type=int, default=456, metavar='N', help='random seed (default: 456)') parser.add_argument('--batch_size', type=int, default=256, metavar='N', help='batch size (default: 256)') -parser.add_argument('--num_steps', type=int, default=500001, metavar='N', +parser.add_argument('--num_steps', type=int, default=1000001, metavar='N', help='maximum number of steps (default: 1000000)') parser.add_argument('--hidden_size', type=int, default=256, metavar='N', help='hidden size (default: 256)')