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

Update config.json

Browse files
Files changed (1) hide show
  1. 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,