fix edvr bug (#320)

244b0225 · wangna11BD · LielinJiang · 1d8cd182 · 244b0225 · 244b0225
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

configs/edvr_l_w_tsa.yaml configs/edvr_l_w_tsa.yaml +1 -0

configs/edvr_m_w_tsa.yaml configs/edvr_m_w_tsa.yaml +1 -0

未找到文件。
--- a/configs/edvr_l_w_tsa.yaml
+++ b/configs/edvr_l_w_tsa.yaml
 total_iters: 600000
 output_dir: output_dir
+find_unused_parameters: True
 checkpoints_dir: checkpoints
 # tensor range for function tensor2img
 min_max:

--- a/configs/edvr_m_w_tsa.yaml
+++ b/configs/edvr_m_w_tsa.yaml
 total_iters: 600000
 output_dir: output_dir
+find_unused_parameters: True
 checkpoints_dir: checkpoints
 # tensor range for function tensor2img
 min_max: