From bf84e5c4dbd952b3301a890ef879d894686dfe87 Mon Sep 17 00:00:00 2001
From: Nisarg Shah <52718252+nisargshah1999@users.noreply.github.com>
Date: Sat, 4 Jul 2020 14:48:50 +0530
Subject: [PATCH] Update extract_features.py

Passing batch size from argparse.ArgumentParser
---
 extract_features.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extract_features.py b/extract_features.py
index a2c2e47..47a2876 100755
--- a/extract_features.py
+++ b/extract_features.py
@@ -31,7 +31,7 @@
 
     args = parser.parse_args()
 
-    params_model = {'bsize': 64, 'word_emb_dim': 300, 'enc_lstm_dim': 2048,
+    params_model = {'bsize': args.batch-size , 'word_emb_dim': 300, 'enc_lstm_dim': 2048,
                     'pool_type': 'max', 'dpout_model': 0.0, 'version': args.version}
     model = InferSent(params_model)
     model.load_state_dict(torch.load(args.model_path))