# i3d_slow_resnet101_f16s4_kinetics700

DDP_CONFIG:
  WORLD_SIZE: 1
  WORLD_RANK: 0
  GPU_WORLD_SIZE: 8
  GPU_WORLD_RANK: 0
  DIST_URL: 'tcp://172.31.69.242:23456'
  WOLRD_URLS: ['172.31.69.242']
  AUTO_RANK_MATCH: True
  DIST_BACKEND: 'nccl'
  GPU: 0
  DISTRIBUTED: True

CONFIG:
  TRAIN:
    EPOCH_NUM: 196
    BATCH_SIZE: 8
    LR: 0.1
    MOMENTUM: 0.9
    W_DECAY: 1e-4
    USE_WARMUP: True
    WARMUP_EPOCHS: 34
    LR_POLICY: 'Cosine'
    RESUME_EPOCH: -1

  VAL:
    FREQ: 5
    BATCH_SIZE: 8

  DATA:
    TRAIN_ANNO_PATH: '/home/ubuntu/data/kinetics700_v1/k700_v1_train.txt'
    VAL_ANNO_PATH: '/home/ubuntu/data/kinetics700_v1/k700_v1_val.txt'
    TRAIN_DATA_PATH: '/home/ubuntu/data/kinetics700_v1/train_256/'
    VAL_DATA_PATH:  '/home/ubuntu/data/kinetics700_v1/val_256/'
    NUM_CLASSES: 700
    CLIP_LEN: 16
    FRAME_RATE: 4
    NUM_SEGMENT: 1
    NUM_CROP: 1
    TEST_NUM_SEGMENT: 10
    TEST_NUM_CROP: 3
    MULTIGRID: False
    KEEP_ASPECT_RATIO: True

  MODEL:
    NAME: 'i3d_slow_resnet101_f16s4_kinetics700'
    PRETRAINED: False

  LOG:
    BASE_PATH: './logs/i3d_slow_resnet101_f16s4_kinetics700'
    LOG_DIR: 'tb_log'
    SAVE_DIR: 'checkpoints'
    EVAL_DIR: './logs/i3d_slow_resnet101_f16s4_kinetics700/eval'
    DISPLAY_FREQ: 50
    SAVE_FREQ: 5