From 244b0225cfcf6e08f4b1615a090a50940421cc49 Mon Sep 17 00:00:00 2001
From: wangna11BD <79366697+wangna11BD@users.noreply.github.com>
Date: Fri, 21 May 2021 22:42:13 +0800
Subject: [PATCH] fix edvr bug (#320)

---
 configs/edvr_l_w_tsa.yaml | 1 +
 configs/edvr_m_w_tsa.yaml | 1 +
 2 files changed, 2 insertions(+)

diff --git a/configs/edvr_l_w_tsa.yaml b/configs/edvr_l_w_tsa.yaml
index 97ce169..9f95deb 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 79a9e4c..c9d5ea8 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:
-- 
GitLab