args management in input call

Author: aditya tewari

examples/lienet/task.py

@@ -1,20 +1,21 @@
 #!/usr/bin/env python3
 import argparse
 import os
+import sys
 import tensorflow as tf
 import numpy as np
 
-from . import model
-from shared import utils
+import model
+from examples.shared import utils
 
 DATA_URL = "https://data.vision.ee.ethz.ch/zzhiwu/ManifoldNetData/LieData/G3D_Lie_data.zip"
 DATA_FOLDER = "lie20_half_inter1"
 G3D_CLASSES = 20
 VAL_SPLIT = 0.2
 
 
-def get_args():
-    parser = argparse.ArgumentParser()
+def get_args(args):
+    parser = argparse.ArgumentParser(args)
     parser.add_argument(
         '--job-dir', type=str, required=True, help='checkpoint dir'
     )
@@ -46,7 +47,7 @@ def get_args():
     return parser.parse_args()
 
 
-def prepare_data():
+def prepare_data(args):
     features, labels = utils.load_matlab_data("fea", args.data_dir, DATA_FOLDER)
     features = np.array([np.stack(example) for example in features.squeeze()])
     # reshape to [batch_size, spatial_dim, temp_dim, num_rows, num_cols]
@@ -61,7 +62,7 @@ def prepare_data():
 
 def train_and_evaluate(args):
     utils.download_data(args.data_dir, DATA_URL, unpack=True)
-    train, val = prepare_data()
+    train, val = prepare_data(args)
 
     train_dataset = (
         tf.data.Dataset.from_tensor_slices(train)
@@ -90,4 +91,5 @@ def train_and_evaluate(args):
 
 if __name__ == "__main__":
     tf.get_logger().setLevel("INFO")
-    train_and_evaluate(get_args())
+    argv = sys.argv[1:]
+    train_and_evaluate(get_args(argv))