Update config.json
Browse files- config.json +24 -1
config.json
CHANGED
@@ -27,17 +27,35 @@
|
|
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
|
40 |
],
|
|
|
41 |
"perceptual_weight": 0.0,
|
42 |
"pool_size": [
|
43 |
4,
|
@@ -49,6 +67,11 @@
|
|
49 |
3,
|
50 |
3
|
51 |
],
|
|
|
|
|
|
|
|
|
|
|
52 |
"strides": [
|
53 |
1,
|
54 |
2,
|
|
|
27 |
7,
|
28 |
7
|
29 |
],
|
30 |
+
"kernel_size_3d": [
|
31 |
+
1,
|
32 |
+
3,
|
33 |
+
3
|
34 |
+
],
|
35 |
+
"kernel_size_3d_discriminator": [
|
36 |
+
3,
|
37 |
+
5,
|
38 |
+
5
|
39 |
+
],
|
40 |
"model_type": "propainter",
|
41 |
"neighbor_length": 10,
|
42 |
"no_dis": false,
|
43 |
+
"norm_fn": [
|
44 |
+
"batch",
|
45 |
+
"group",
|
46 |
+
"instance",
|
47 |
+
"none"
|
48 |
+
],
|
49 |
"num_attention_heads": 4,
|
50 |
"num_channels": 128,
|
51 |
"num_hidden_layers": 8,
|
52 |
"num_local_frames_propainter": 10,
|
53 |
+
"padding": 1,
|
54 |
+
"padding_inpaint_generator": [
|
55 |
3,
|
56 |
3
|
57 |
],
|
58 |
+
"patch_size": 3,
|
59 |
"perceptual_weight": 0.0,
|
60 |
"pool_size": [
|
61 |
4,
|
|
|
67 |
3,
|
68 |
3
|
69 |
],
|
70 |
+
"stride_3d": [
|
71 |
+
1,
|
72 |
+
1,
|
73 |
+
1
|
74 |
+
],
|
75 |
"strides": [
|
76 |
1,
|
77 |
2,
|