File size: 1,042 Bytes
0c51b93 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | 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
|