fix edvr train on rocm (#331)

60c41c2f · LielinJiang · GitHub · 870c902a · 60c41c2f
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppgan/models/edvr_model.py ppgan/models/edvr_model.py +1 -1

未找到文件。
--- a/ppgan/models/edvr_model.py
+++ b/ppgan/models/edvr_model.py
@@ -48,7 +48,7 @@ class EDVRModel(BaseSRModel):
        self.visual_items['lq+1'] = self.lq[:, 3, :, :, :]
        self.visual_items['lq+2'] = self.lq[:, 4, :, :, :]
        if 'gt' in input:
-            self.gt = paddle.to_tensor(input['gt'])
+            self.gt = input['gt']
            self.visual_items['gt'] = self.gt
        self.image_paths = input['lq_path']