Update vqa_preprocessing.py

GT-Vision-Lab · Dec 23, 2015 · 747950d · 747950d
1 parent 99d766f
commit 747950d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data/vqa_preprocessing.py b/data/vqa_preprocessing.py
@@ -30,7 +30,7 @@ def download_vqa():
 
 
 def main(params):
-    if params['download'] == True:
+    if params['download'] == 'True':
         download_vqa()
 
     '''
@@ -122,7 +122,7 @@ def main(params):
     parser = argparse.ArgumentParser()
 
     # input json
-    parser.add_argument('--download', default=False, help='Download and extract data from VQA server')
+    parser.add_argument('--download', default='False', help='Download and extract data from VQA server')
     parser.add_argument('--split', default=1, type=int, help='1: train on Train and test on Val, 2: train on Train+Val and test on Test')
 
     args = parser.parse_args()