Hugging Face's logo

Spaces:
SuyeonJ
/
VfiTest
Sleeping

App Files Community
VfiTest
File size: 1,342 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
86
 
exp_name: FET-VFI_BiFrequencyLoss

mode: train

train_dataset:
  name: vimeo
  args:
    root_path: ../datasets/frames1_triplet
    flow_root: ../datasets/frames1_unimatch_flow
    tri_trainlist: tri_trainlist.txt
    split: train
    patch_size: 256
    flow: 't0'
  loader:
    batch_size: 16
    num_workers: 4

# 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: 4.e-6, weight_decay: 1.e-4}

lr_scheduler:
  name: one_cycle_lr
  args:
    max_lr: 4.e-6
    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 }
  }

pretrained: ./pretrained/upr_freq002.pth
max_epoch: 300

validate_every: 10
save_every: 10
vis_every: 20

seed: 1234

dist_url: env://