|
TRAIN: |
|
ENABLE: True |
|
DATASET: ssv2 |
|
BATCH_SIZE: 16 |
|
EVAL_PERIOD: 5 |
|
CHECKPOINT_PERIOD: 5 |
|
AUTO_RESUME: True |
|
DATA: |
|
PATH_TO_DATA_DIR: " /path/to/ssv2/annotations/" |
|
PATH_PREFIX: "/path/to/ssv2/frames/" |
|
NUM_FRAMES: 64 |
|
SAMPLING_RATE: 2 |
|
TRAIN_JITTER_SCALES: [256, 320] |
|
TRAIN_CROP_SIZE: 224 |
|
TEST_CROP_SIZE: 256 |
|
INPUT_CHANNEL_NUM: [3, 3] |
|
INV_UNIFORM_SAMPLE: True |
|
RANDOM_FLIP: False |
|
REVERSE_INPUT_CHANNEL: True |
|
SLOWFAST: |
|
ALPHA: 4 |
|
BETA_INV: 8 |
|
FUSION_CONV_CHANNEL_RATIO: 2 |
|
FUSION_KERNEL_SZ: 7 |
|
RESNET: |
|
SPATIAL_STRIDES: [[1, 1], [2, 2], [2, 2], [2, 2]] |
|
SPATIAL_DILATIONS: [[1, 1], [1, 1], [1, 1], [1, 1]] |
|
ZERO_INIT_FINAL_BN: True |
|
WIDTH_PER_GROUP: 64 |
|
NUM_GROUPS: 1 |
|
DEPTH: 50 |
|
TRANS_FUNC: bottleneck_transform |
|
STRIDE_1X1: False |
|
NUM_BLOCK_TEMP_KERNEL: [[3, 3], [4, 4], [6, 6], [3, 3]] |
|
NONLOCAL: |
|
LOCATION: [[[], []], [[], []], [[], []], [[], []]] |
|
GROUP: [[1, 1], [1, 1], [1, 1], [1, 1]] |
|
INSTANTIATION: dot_product |
|
BN: |
|
USE_PRECISE_STATS: True |
|
NUM_BATCHES_PRECISE: 200 |
|
NORM_TYPE: sync_batchnorm |
|
NUM_SYNC_DEVICES: 4 |
|
SOLVER: |
|
BASE_LR: 0.2 |
|
LR_POLICY: cosine |
|
MAX_EPOCH: 200 |
|
MOMENTUM: 0.9 |
|
WEIGHT_DECAY: 1e-4 |
|
WARMUP_EPOCHS: 34.0 |
|
WARMUP_START_LR: 0.01 |
|
OPTIMIZING_METHOD: sgd |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
MODEL: |
|
NUM_CLASSES: 174 |
|
ARCH: slowfast |
|
LOSS_FUNC: cross_entropy |
|
DROPOUT_RATE: 0.5 |
|
TEST: |
|
ENABLE: True |
|
DATASET: ssv2 |
|
BATCH_SIZE: 16 |
|
NUM_ENSEMBLE_VIEWS: 1 |
|
NUM_SPATIAL_CROPS: 1 |
|
DATA_LOADER: |
|
NUM_WORKERS: 4 |
|
PIN_MEMORY: True |
|
NUM_GPUS: 8 |
|
NUM_SHARDS: 1 |
|
RNG_SEED: 0 |
|
OUTPUT_DIR: . |
|
|
|
LOG_MODEL_INFO: True |
|
|
