File size: 1,204 Bytes
8d015d4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
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
|