| TRAIN: | |
| ENABLE: False | |
| DATASET: kinetics | |
| BATCH_SIZE: 16 | |
| EVAL_PERIOD: 1 | |
| CHECKPOINT_PERIOD: 2 | |
| AUTO_RESUME: True | |
| CHECKPOINT_FILE_PATH: your_path_to_checkpoint | |
| FINETUNE: True | |
| DATA: | |
| PATH_TO_DATA_DIR: /path/to/BASKET/ | |
| NUM_FRAMES: 96 | |
| SAMPLING_RATE: 156 | |
| TRAIN_JITTER_SCALES: [224, 224] | |
| TRAIN_CROP_SIZE: 224 | |
| TEST_CROP_SIZE: 224 | |
| INPUT_CHANNEL_NUM: [3] | |
| PATH_LABEL_SEPARATOR: ',' | |
| MULTI_LABEL: True | |
| TIMESFORMER: | |
| ATTENTION_TYPE: 'divided_space_time' | |
| SOLVER: | |
| BASE_LR: 0.1 | |
| LR_POLICY: steps_with_relative_lrs | |
| STEPS: [0, 3, 4] | |
| LRS: [1, 0.1, 0.01] | |
| MAX_EPOCH: 5 | |
| MOMENTUM: 0.9 | |
| WEIGHT_DECAY: 1e-4 | |
| OPTIMIZING_METHOD: sgd | |
| MODEL: | |
| MODEL_NAME: vit_base_patch16_224 | |
| NUM_CLASSES: 5 | |
| ARCH: vit | |
| LOSS_FUNC: cross_entropy | |
| DROPOUT_RATE: 0.5 | |
| TEST: | |
| ENABLE: True | |
| DATASET: kinetics | |
| BATCH_SIZE: 8 | |
| NUM_ENSEMBLE_VIEWS: 1 | |
| NUM_SPATIAL_CROPS: 1 | |
| DATA_LOADER: | |
| NUM_WORKERS: 1 | |
| PIN_MEMORY: True | |
| TENSORBOARD: | |
| ENABLE: True | |
| MODEL_VIS: | |
| ENABLE: True | |
| NUM_GPUS: 8 | |
| NUM_SHARDS: 1 | |
| RNG_SEED: 0 | |
| OUTPUT_DIR: output/BASKET | |