Update config.json

config.json

@@ -6,8 +6,8 @@
   ],
   "batch_size_flow_complete_net": 8,
   "batch_size_propainter": 8,
-  "beta1_flow_complete_net": 0,
-  "beta1_propainter": 0,
+  "beta1_flow_complete_net": 0.0,
+  "beta1_propainter": 0.0,
   "beta2_flow_complete_net": 0.99,
   "beta2_propainter": 0.99,
   "channels": [
@@ -17,14 +17,14 @@
   ],
   "corr_levels": 4,
   "corr_radius": 4,
-  "dropout": 0,
+  "dropout": 0.0,
   "flow_weight_flow_complete_net": 0.25,
-  "flow_weight_propainter": 1,
+  "flow_weight_propainter": 1.0,
   "gan_loss": "hinge",
   "height_flow_complete_net": 240,
   "height_propainter": 240,
   "hidden_size": 512,
-  "hole_weight": 1,
+  "hole_weight": 1.0,
   "in_channels": [
     64,
     64,
@@ -32,21 +32,21 @@
   ],
   "initializer_range": 0.02,
   "interp_mode": "nearest",
-  "iterations_flow_complete_net": 700000.0,
-  "iterations_propainter": 700000.0,
+  "iterations_flow_complete_net": 700000,
+  "iterations_propainter": 700000,
   "kernel_size": [
     7,
     7
   ],
-  "load_flow_flow_complete_net": 0,
-  "load_flow_propainter": 0,
+  "load_flow_in_flow_complete_net": false,
+  "load_flow_in_propainter": false,
   "log_freq_flow_complete_net": 100,
   "log_freq_propainter": 100,
   "lr_flow_complete_net": 5e-05,
   "lr_propainter": 0.0001,
   "model_type": "propainter",
   "neighbor_length": 10,
-  "no_dis": 0,
+  "no_dis": false,
   "num_attention_heads": 4,
   "num_channels": 128,
   "num_hidden_layers": 8,
@@ -62,7 +62,7 @@
     3,
     3
   ],
-  "perceptual_weight": 0,
+  "perceptual_weight": 0.0,
   "pool_size": [
     4,
     4
@@ -70,8 +70,8 @@
   "raft_iter": 20,
   "raft_optical_flow_iter": 20,
   "ref_stride": 10,
-  "save_freq_flow_complete_net": 5000.0,
-  "save_freq_propainte": 10000.0,
+  "save_freq_flow_complete_net": 5000,
+  "save_freq_propainter": 10000,
   "stride": [
     3,
     3
@@ -83,7 +83,7 @@
   ],
   "subvideo_length": 80,
   "transformers_version": "4.45.0.dev0",
-  "valid_weight": 1,
+  "valid_weight": 1.0,
   "width_flow_complete_net": 432,
   "width_propainter": 432,
   "window_size": [