diff --git a/configs/edvr_l_w_tsa.yaml b/configs/edvr_l_w_tsa.yaml
index 97ce169f3e569e31bcca0e778e8b1dcf3f47a82c..9f95deb7e68c74a01568ddd7b929c4576af51e80 100644
--- a/configs/edvr_l_w_tsa.yaml
+++ b/configs/edvr_l_w_tsa.yaml
@@ -1,5 +1,6 @@
 total_iters: 600000
 output_dir: output_dir
+find_unused_parameters: True
 checkpoints_dir: checkpoints
 # tensor range for function tensor2img
 min_max:
diff --git a/configs/edvr_m_w_tsa.yaml b/configs/edvr_m_w_tsa.yaml
index 79a9e4cf6fa68b052107892ed6b9582cc469f66f..c9d5ea8c7aefe79e1ace00d0fa8e115577048e1a 100644
--- a/configs/edvr_m_w_tsa.yaml
+++ b/configs/edvr_m_w_tsa.yaml
@@ -1,5 +1,6 @@
 total_iters: 600000
 output_dir: output_dir
+find_unused_parameters: True
 checkpoints_dir: checkpoints
 # tensor range for function tensor2img
 min_max: