Skip to content

Commit

Permalink
Update eval_with_train_config
Browse files Browse the repository at this point in the history
  • Loading branch information
Ironieser authored Mar 15, 2024
1 parent 0760da1 commit 3bf4e25
Showing 1 changed file with 41 additions and 0 deletions.
41 changes: 41 additions & 0 deletions configs/eval_with_train_config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
TRAIN:
BATCH_SIZE: 8
DROPOUT: 0.0


MODEL:
DIM_EMBEDDING: 128
TRANSFORMER: True
ALIGNMENT: True


DATASET:
MODE: 'test'


## CSV dataset
# NAME: 'CSV'
# TXT_PATH: '/storage/data/dongsx/dataset/CSV/splits/test_pairs.txt'
# NUM_CLASS: 45

# COIN dataset
# NAME: 'COIN-SV'
# TXT_PATH: './Datasets/COIN-SV/val_pairs.txt'
# TXT_PATH: '/storage/data/dongsx/dataset/COIN/splits/val_pairs.txt'
# TXT_PATH: '/storage/data/dongsx/dataset/COIN/splits/test_pairs.txt'
# TXT_PATH: './Datasets/COIN-SV/test_pairs.txt'
# NUM_CLASS: 37

# Diving48 dataset;
# NAME: 'DIVING48-SV'
# TXT_PATH: '/Datasets/DIVING48-SV/val_pairs.txt'
# TXT_PATH: '/Datasets/DIVING48-SV/test_pairs.txt'
# NUM_CLASS: 20


NUM_CLIP: 16
AUGMENT: False
SHUFFLE: False
NUM_WORKERS: 10


0 comments on commit 3bf4e25

Please sign in to comment.