English
zarus03 commited on
Commit
e2eafc5
·
verified ·
1 Parent(s): a3c8a6a

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. demo.py +6 -0
demo.py CHANGED
@@ -21,6 +21,12 @@ args_parser = argparse.ArgumentParser()
21
  args_parser.add_argument("--datatype", type=str, default="msvd", help="dataset name")
22
  args_parser.add_argument("--local-rank", default=0, type=int, help="distribted training")
23
 
 
 
 
 
 
 
24
  args = args_parser.parse_args()
25
 
26
  logger = logging.getLogger(__name__)
 
21
  args_parser.add_argument("--datatype", type=str, default="msvd", help="dataset name")
22
  args_parser.add_argument("--local-rank", default=0, type=int, help="distribted training")
23
 
24
+ args_parser.add_argument('--train_csv', type=str, default='data/.train.csv', help='')
25
+ args_parser.add_argument('--val_csv', type=str, default='data/.val.csv', help='')
26
+ args_parser.add_argument('--data_path', type=str, default='data/caption.pickle', help='data pickle file path')
27
+ args_parser.add_argument('--features_path', type=str, default='data/videos_feature.pickle', help='feature path')
28
+
29
+
30
  args = args_parser.parse_args()
31
 
32
  logger = logging.getLogger(__name__)