Update main.py

This commit is contained in:
Pranjal Tandon
2019-01-20 16:51:11 +05:30
committed by GitHub
parent aeb7d40a6e
commit 1423a10727
+1 -1
View File
@@ -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)')