linear_R3D-18_syn_anyckpt_100e_lr0.1.yaml

TASK: ssl_eval_ucf
TRAIN:
  ENABLE: True
  DATASET: kinetics
  BATCH_SIZE: 32
  EVAL_PERIOD: 10
  CHECKPOINT_PERIOD: 10
  AUTO_RESUME: True
  CHECKPOINT_CLEAR_NAME_PATTERN: ("backbone.",)
  CHECKPOINT_EPOCH_RESET: True
  #CHECKPOINT_FILE_PATH: './checkpoints/ssl_checkpoint_epoch_00500.pyth'
RESNET:
  DEPTH: 18
  NUM_BLOCK_TEMP_KERNEL: [[2], [2], [2], [2]]
  TRANS_FUNC: basic_transform_r3d
  STEM_FUNC: r3d_stem
DATA:
  NUM_FRAMES: 16
  SAMPLING_RATE: 4
  SSL_COLOR_JITTER: True # default false
  COLOR_RND_GRAYSCALE: 0.2 # default 0.0
  SSL_COLOR_HUE: 0.1
  SSL_COLOR_BRI_CON_SAT: [0.4, 0.4, 0.4] # default [0.4, 0.4, 0.4]
  TRAIN_JITTER_SCALES: [128, 160]
  TRAIN_CROP_SIZE: 112
  TEST_CROP_SIZE: 128
  INPUT_CHANNEL_NUM: [3]
  DECODING_BACKEND: pyav
  PATH_TO_DATA_DIR: /disk/gao2/code/VideoMAE/data/ucf101/
BN:
  USE_PRECISE_STATS: False
  NUM_BATCHES_PRECISE: 200
  WEIGHT_DECAY: 0.0
  NUM_SYNC_DEVICES: 1
  NORM_TYPE: "sync_batchnorm"
SOLVER:
  BASE_LR: 0.1
  BASE_LR_SCALE_NUM_SHARDS: True
  LR_POLICY: cosine
  MAX_EPOCH: 100
  MOMENTUM: 0.9
  WEIGHT_DECAY: 0.0
  WARMUP_EPOCHS: 5.0
  WARMUP_START_LR: 0.0001
  OPTIMIZING_METHOD: sgd
MODEL:
  NUM_CLASSES: 101
  MODEL_NAME: ResNet
  ARCH: r3d_shallow
  LOSS_FUNC: cross_entropy
  DROPOUT_RATE: 0.0
  DETACH_FINAL_FC: True
TEST:
  ENABLE: True
  DATASET: kinetics
  BATCH_SIZE: 64
DATA_LOADER:
  NUM_WORKERS: 8
  PIN_MEMORY: True
MIXUP:
  ENABLE: False
NUM_GPUS: 1
NUM_SHARDS: 1
RNG_SEED: 0
OUTPUT_DIR: ./work_dirs/finetune_ucf101/