# i3d_resnet50_v1_sthsthv2

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

CONFIG:
  TRAIN:
    EPOCH_NUM: 100
    BATCH_SIZE: 8
    LR: 0.01
    LR_POLICY: 'Step'
    MOMENTUM: 0.9
    W_DECAY: 1e-5
    USE_WARMUP: False
    LR_MILESTONE: [40, 80]
    STEP: 0.1

  VAL:
    FREQ: 2
    BATCH_SIZE: 8

  DATA:
    TRAIN_ANNO_PATH: '/home/ubuntu/data/sthsthv2/sthsthv2_train.txt'
    VAL_ANNO_PATH: '/home/ubuntu/data/sthsthv2/sthsthv2_val.txt'
    TRAIN_DATA_PATH: '/home/ubuntu/data/sthsthv2/20bn-something-something-v2/'
    VAL_DATA_PATH:  '/home/ubuntu/data/sthsthv2/20bn-something-something-v2/'
    NUM_CLASSES: 174
    CLIP_LEN: 16
    FRAME_RATE: 2
    NUM_SEGMENT: 1
    NUM_CROP: 1
    TEST_NUM_SEGMENT: 2
    TEST_NUM_CROP: 3
    MULTIGRID: False
    KEEP_ASPECT_RATIO: False

  MODEL:
    NAME: 'i3d_resnet50_v1_sthsthv2'
    PRETRAINED: False

  LOG:
    BASE_PATH: './logs/i3d_resnet50_v1_sthsthv2'
    LOG_DIR: 'tb_log'
    SAVE_DIR: 'checkpoints'
    EVAL_DIR: './logs/i3d_resnet50_v1_sthsthv2/eval'
    SAVE_FREQ: 2