From 5dfb507f36b1d59a1d32e84ed75001d6920a48d6 Mon Sep 17 00:00:00 2001 From: Richard Zou Date: Mon, 24 Sep 2018 08:10:40 -0700 Subject: [PATCH] Fix miniBatch size in profile.py --- rnns/fastrnns/profile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)