File size: 1,014 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 |
exp_name: backwarp_resize
mode: train
train_dataset:
name: vimeo
args:
root_path: /mnt/nas/512_new
split: train
patch_size: 512
flow: none
loader:
batch_size: 16
num_workers: 8
test_dataset:
name: vimeo
args:
root_path: /mnt/nas/vimeo_triplet
split: val
loader:
batch_size: 16
num_workers: 8
demo_dataset:
name: demo
args:
root_path: ../data/animation
split: animation
model:
name: upr_net_mod
args:
pyr_level: 5
nr_lvl_skipped: 0
splat_mode: average
optimizer:
name: adamW
args: {lr: 0.000075, weight_decay: 1.e-4}
lr_scheduler:
name: one_cycle_lr
args:
max_lr: 0.000075
pct_start: 0.01
cycle_momentum: False
anneal_strategy: cos
loss:
- {
name: charbonnier,
args: { weight: 1 }
}
- {
name: ternary,
args: { weight: 1 }
}
max_epoch: 533
validate_every: 1
save_every: 10
vis_every: 20
seed: 1234
dist_url: env://
pretrained: ./pretrained/upr_backwarp_corr_resize_model.pth
|