diff --git a/rnns/fastrnns/profile.py b/rnns/fastrnns/profile.py index 65c7d746b8..b76fa4fdb3 100644 --- a/rnns/fastrnns/profile.py +++ b/rnns/fastrnns/profile.py @@ -106,7 +106,7 @@ def full_profile(rnns, **args): parser.add_argument('--numLayers', default='1', type=int) parser.add_argument('--inputSize', default='512', type=int) parser.add_argument('--hiddenSize', default='512', type=int) - parser.add_argument('--miniBatch', default='512', type=int) + parser.add_argument('--miniBatch', default='64', type=int) parser.add_argument('--sleep_between_seconds', default='1', type=int) parser.add_argument('--nloops', default='5', type=int)