raft_kubric_improved_aug.yaml

_BASE_: "../_base_/kubric_improved_aug.yaml"
DATA:
  BATCH_SIZE: 10
  NORM_PARAMS: {"use": True, "mean":[127.5, 127.5, 127.5], "std":[127.5, 127.5, 127.5]}
  APPEND_VALID_MASK: True
  TRAIN_DATASET:
    Kubric:
      CROP: 
        USE: True
        SIZE: [368, 496]
        TYPE: "random"
  VAL_DATASET:
    Kubric: 
      PADDING: 1
      CROP: 
        USE: True
        SIZE: [368, 496]
        TYPE: "center"
SCHEDULER:
  USE: True
  NAME: OneCycleLR
  PARAMS:
    max_lr: 0.0004
    total_steps: 100100
    pct_start: 0.05
    cycle_momentum: False
    anneal_strategy: linear
CRITERION:
  CUSTOM: True
  NAME: SequenceLoss
  PARAMS: 
    gamma: 0.8
    max_flow: 400.0
EPOCHS: null
NUM_STEPS: 100100
LOG_DIR: "./logs"
CKPT_DIR: "./ckpts"
LOG_ITERATIONS_INTERVAL: 100
CKPT_INTERVAL: 20000
VALIDATE_INTERVAL: 1000
VALIDATE_ON: metric