ruffy369 commited on
Commit
096325d
·
verified ·
1 Parent(s): 59f1f7d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +0 -29
config.json CHANGED
@@ -4,12 +4,6 @@
4
  "architectures": [
5
  "ProPainterModel"
6
  ],
7
- "batch_size_flow_complete_net": 8,
8
- "batch_size_propainter": 8,
9
- "beta1_flow_complete_net": 0.0,
10
- "beta1_propainter": 0.0,
11
- "beta2_flow_complete_net": 0.99,
12
- "beta2_propainter": 0.99,
13
  "channels": [
14
  64,
15
  96,
@@ -19,10 +13,7 @@
19
  "corr_radius": 4,
20
  "dropout": 0.0,
21
  "flow_weight_flow_complete_net": 0.25,
22
- "flow_weight_propainter": 1.0,
23
  "gan_loss": "hinge",
24
- "height_flow_complete_net": 240,
25
- "height_propainter": 240,
26
  "hidden_size": 512,
27
  "hole_weight": 1.0,
28
  "in_channels": [
@@ -32,32 +23,17 @@
32
  ],
33
  "initializer_range": 0.02,
34
  "interp_mode": "nearest",
35
- "iterations_flow_complete_net": 700000,
36
- "iterations_propainter": 700000,
37
  "kernel_size": [
38
  7,
39
  7
40
  ],
41
- "load_flow_in_flow_complete_net": false,
42
- "load_flow_in_propainter": false,
43
- "log_freq_flow_complete_net": 100,
44
- "log_freq_propainter": 100,
45
- "lr_flow_complete_net": 5e-05,
46
- "lr_propainter": 0.0001,
47
  "model_type": "propainter",
48
  "neighbor_length": 10,
49
  "no_dis": false,
50
  "num_attention_heads": 4,
51
  "num_channels": 128,
52
  "num_hidden_layers": 8,
53
- "num_local_frames_flow_complete_net": 10,
54
  "num_local_frames_propainter": 10,
55
- "num_prefetch_queue_flow_complete_net": 4,
56
- "num_prefetch_queue_propainter": 8,
57
- "num_ref_frames_flow_complete_net": 1,
58
- "num_ref_frames_propainter": 6,
59
- "num_workers_flow_complete_net": 4,
60
- "num_workers_propainter": 8,
61
  "padding": [
62
  3,
63
  3
@@ -68,10 +44,7 @@
68
  4
69
  ],
70
  "raft_iter": 20,
71
- "raft_optical_flow_iter": 20,
72
  "ref_stride": 10,
73
- "save_freq_flow_complete_net": 5000,
74
- "save_freq_propainter": 10000,
75
  "stride": [
76
  3,
77
  3
@@ -84,8 +57,6 @@
84
  "subvideo_length": 80,
85
  "transformers_version": "4.45.0.dev0",
86
  "valid_weight": 1.0,
87
- "width_flow_complete_net": 432,
88
- "width_propainter": 432,
89
  "window_size": [
90
  5,
91
  9
 
4
  "architectures": [
5
  "ProPainterModel"
6
  ],
 
 
 
 
 
 
7
  "channels": [
8
  64,
9
  96,
 
13
  "corr_radius": 4,
14
  "dropout": 0.0,
15
  "flow_weight_flow_complete_net": 0.25,
 
16
  "gan_loss": "hinge",
 
 
17
  "hidden_size": 512,
18
  "hole_weight": 1.0,
19
  "in_channels": [
 
23
  ],
24
  "initializer_range": 0.02,
25
  "interp_mode": "nearest",
 
 
26
  "kernel_size": [
27
  7,
28
  7
29
  ],
 
 
 
 
 
 
30
  "model_type": "propainter",
31
  "neighbor_length": 10,
32
  "no_dis": false,
33
  "num_attention_heads": 4,
34
  "num_channels": 128,
35
  "num_hidden_layers": 8,
 
36
  "num_local_frames_propainter": 10,
 
 
 
 
 
 
37
  "padding": [
38
  3,
39
  3
 
44
  4
45
  ],
46
  "raft_iter": 20,
 
47
  "ref_stride": 10,
 
 
48
  "stride": [
49
  3,
50
  3
 
57
  "subvideo_length": 80,
58
  "transformers_version": "4.45.0.dev0",
59
  "valid_weight": 1.0,
 
 
60
  "window_size": [
61
  5,
62
  9