Update config.json

config.json

@@ -18,11 +18,16 @@
   "flow_weight_propainter": 1,
   "height_flow_complete_net": 240,
   "height_propainter": 240,
+  "hidden_size": 512,
   "hole_weight": 1,
   "initializer_range": 0.02,
   "interp_mode": "nearest",
   "iterations_flow_complete_net": 700000.0,
   "iterations_propainter": 700000.0,
+  "kernel_size": [
+    7,
+    7
+  ],
   "load_flow_flow_complete_net": 0,
   "load_flow_propainter": 0,
   "log_freq_flow_complete_net": 100,
@@ -31,6 +36,9 @@
   "lr_propainter": 0.0001,
   "model_type": "propainter",
   "neighbor_length": 10,
+  "num_attention_heads": 4,
+  "num_channels": 128,
+  "num_hidden_layers": 8,
   "num_local_frames_flow_complete_net": 10,
   "num_local_frames_propainter": 10,
   "num_prefetch_queue_flow_complete_net": 4,
@@ -39,15 +47,31 @@
   "num_ref_frames_propainter": 6,
   "num_workers_flow_complete_net": 4,
   "num_workers_propainter": 8,
+  "padding": [
+    3,
+    3
+  ],
   "perceptual_weight": 0,
+  "pool_size": [
+    4,
+    4
+  ],
   "raft_iter": 20,
   "raft_optical_flow_iter": 20,
   "ref_stride": 10,
   "save_freq_flow_complete_net": 5000.0,
   "save_freq_propainte": 10000.0,
+  "stride": [
+    3,
+    3
+  ],
   "subvideo_length": 80,
   "transformers_version": "4.45.0.dev0",
   "valid_weight": 1,
   "width_flow_complete_net": 432,
-  "width_propainter": 432
+  "width_propainter": 432,
+  "window_size": [
+    5,
+    9
+  ]
 }