Upload folder using huggingface_hub
Browse files
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__)
|