add configs of downstream task

configs/eval_config_retrival.yml

@@ -0,0 +1,34 @@
+TRAIN:
+
+  MAX_EPOCH: 300
+  BATCH_SIZE: 4
+
+  LR: 0.0005
+  DROPOUT: 0.0
+  SEED: 1234
+  FREEZE_BACKBONE: True
+
+  SAVE_PATH: 'retrival_logs'
+
+MODEL:
+
+  PRETRAIN: '/public/home/dongsx/svip/pretrained_models/resnet50-kinetics400.pth'
+
+
+DATASET:
+  MODE: 'test'
+
+#  CSV dataset
+  NAME: 'CSV'
+  TXT_PATH: '/storage/data/dongsx/dataset/CSV/splits/text_retrieval.txt'
+  NUM_CLASS: 45
+
+
+  NUM_CLIP: 16
+  AUGMENT: False
+  SHUFFLE: False
+  NUM_WORKERS: 10
+
+
+
+

configs/train_config_cls.yml

@@ -0,0 +1,64 @@
+TRAIN:
+  USE_CUDA: True
+  MAX_EPOCH: 300
+  BATCH_SIZE: 8
+# ddp
+#  BATCH_SIZE: 4
+
+  LR: 0.0005
+  DROPOUT: 0.0
+  SEED: 1234
+  FREEZE_BACKBONE: False
+
+  SAVE_PATH: 'csv_logs'
+#  SAVE_PATH: 'coin_logs'
+#  SAVE_PATH: 'diving_logs'
+
+MODEL:
+
+  BACKBONE: 'resnet'
+
+
+#  PRETRAIN: '/public/home/dongsx/svip/pretrained_models/tsn_r50_256p_1x1x3_100e_kinetics400_rgb_20200725-22592236.pth'
+#
+  PRETRAIN: '/public/home/dongsx/svip/pretrained_models/resnet50-kinetics400.pth'
+  DIM_EMBEDDING: 128
+  TRANSFORMER: True
+  ALIGNMENT: True
+  SEQ_LOSS_COEF: 0.0
+  INFO_LOSS_COEF: 0.0
+  GUMBEL_LOSS_COEF: 0.0
+  SAVE_EPOCHS: 100
+
+
+DATASET:
+  MODE: 'train'
+  RANDOM_SAMPLE: True
+
+#  CSV dataset
+  NAME: 'CSV'
+  TXT_PATH: '/storage/data/dongsx/dataset/CSV/splits/transfer_train.txt'
+#  TXT_PATH: '/storage/data/dongsx/dataset/CSV/splits/train_pair.txt'
+  NUM_CLASS: 25
+
+#  COIN dataset
+#  NAME: 'COIN-SV'
+#  TXT_PATH: '/storage/data/dongsx/dataset/COIN/splits/train_pairs.txt'
+#  NUM_CLASS: 37
+
+#  Diving dataset
+#  NAME: 'DIVING48-SV'
+#  TXT_PATH: './Datasets/DIVING48-SV/train_pairs.txt'
+#  NUM_CLASS: 20
+
+
+
+  NUM_SAMPLE: 680
+  NUM_CLIP: 16
+  AUGMENT: True
+  SHUFFLE: True
+  NUM_WORKERS: 10
+
+
+
+