bug fix argparser

distribute.py

@@ -1,5 +1,5 @@
 # edited from https://github.com/fastai/imagenet-fast/blob/master/imagenet_nv/distributed.py
-import os
+import os, sys
 import math
 import time
 import subprocess
@@ -130,7 +130,7 @@ def main():
         type=str,
         help='Training output folder to continue training. Use to continue a training. If it is used, "config_path" is ignored.',
         default='',
-        required=True)
+        required='--config_path' not in sys.argv)
     parser.add_argument(
         '--restore_path',
         type=str,

layers/common_layers.py

@@ -226,6 +226,13 @@ class Attention(nn.Module):
         return alpha
 
     def forward(self, query, inputs, processed_inputs, mask):
+        """
+        shapes:
+            query: B x D_attn_rnn
+            inputs: B x T_en x D_en
+            processed_inputs:: B x T_en x D_attn
+            mask: B x T_en
+        """
         if self.location_attention:
             attention, _ = self.get_location_attention(
                 query, processed_inputs)

train.py

@@ -649,7 +649,7 @@ if __name__ == '__main__':
         type=str,
         help='Training output folder to continue training. Use to continue a training. If it is used, "config_path" is ignored.',
         default='',
-        required=True)
+        required='--config_path' not in sys.argv)
     parser.add_argument(
         '--restore_path',
         type=str,