cfgs/upr_backward_unimatch_exp2.yaml

exp_name: backwarp_resize

mode: train

train_dataset:
  name: vimeo
  args:
    root_path: ../datasets/Vimeo90Kx2
    flow_root: ../datasets/unimatch_flow
    split: train
    patch_size: 512
    flow: t0
  loader:
    batch_size: 16
    num_workers: 16

test_dataset:
  name: vimeo
  args:
    root_path: ../datasets/Vimeo90K
    split: val
  loader:
    batch_size: 32
    num_workers: 8

demo_dataset:
  name: demo
  args:
    root_path: ../data/animation
    split: animation

model:
  name: upr_net_mod
  args:
    pyr_level: 3
    nr_lvl_skipped: 0
    splat_mode: average


optimizer:
  name: adamW
  args: {lr: 0.00015, weight_decay: 1.e-4}

lr_scheduler:
  name: one_cycle_lr
  args:
    max_lr: 0.00015
    pct_start: 0.01
    cycle_momentum: False
    anneal_strategy: cos

loss:
  - {
    name: multiple_flow,
    args: { weight: 0.005 }
  }
  - {
    name: charbonnier,
    args: { weight: 1 }
  }
  - {
    name: ternary,
    args: { weight: 1 }
  }

max_epoch: 250

validate_every: 5
save_every: 10
vis_every: 40

seed: 1234

dist_url: env://
resume: ./save/upr_backward_unimatch_exp2_backwarp_resize/checkpoints/model_30.pth
# pretrained: ./pretrained/upr_backwarp_corr_resize_model.pth