File size: 1,224 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 78 79 80 81 82 83 84 85 |
exp_name: FET-VFI_BiFrequencyLoss
mode: train
train_dataset:
name: vimeo
args:
root_path: ../datasets/Vimeo90K
flow_root: ../datasets/unimatch_flow
split: train
patch_size: 256
flow: 't0'
loader:
batch_size: 32
num_workers: 8
test_dataset:
name: vimeo
args:
root_path: ../datasets/Vimeo90K
split: val
loader:
batch_size: 16
num_workers: 4
save_imgs: False
demo_dataset:
name: demo
args:
root_path: ../data/animation
split: animation
model:
name: upr_net_freq
args:
pyr_level: 3
nr_lvl_skipped: 0
splat_mode: average
fftshift: False
optimizer:
name: adamW
args: {lr: 1.5e-4, weight_decay: 1.e-4}
lr_scheduler:
name: one_cycle_lr
args:
max_lr: 1.5e-4
pct_start: 0.01
cycle_momentum: False
anneal_strategy: cos
loss:
- {
name: frequency,
args: { weight: 0.01 }
}
- {
name: bi_frequency,
args: { weight: 0.01 }
}
- {
name: multiple_flow,
args: { weight: 0.005 }
}
- {
name: charbonnier,
args: { weight: 1 }
}
- {
name: ternary,
args: { weight: 1 }
}
max_epoch: 540
validate_every: 10
save_every: 10
vis_every: 20
seed: 1234
dist_url: env:// |