upadate imagenetNPU

read_imagenet.py

@@ -59,8 +59,13 @@ def UploadToQizhi(train_dir, obs_train_url):
             EnvToObs(train_dir, obs_train_url)
     return
 
-if __name__ == "__main__":
+parser = argparse.ArgumentParser(description='Read big dataset ImageNet Example')
+parser.add_argument('--train_url',
+                    help='output folder to save/load',
+                    default= '/cache/output/')
 
+if __name__ == "__main__":
+    args, unknown = parser.parse_known_args()
     data_path = '/cache/sfs/data/imagenet/'
     modelart_output = '/cache/output'
     if not os.path.exists(modelart_output):
@@ -85,4 +90,4 @@ if __name__ == "__main__":
 
     data_info = dataset_train.to_json(filename= modelart_output + '/data_info.json')
     print(data_info)
-    UploadToQizhi(modelart_output, '/cache/output/')
+    UploadToQizhi(modelart_output, args.train_url)