fix edvr train on rocm

5da06e50 · LielinJiang · d401a15e · 5da06e50
显示空白变更内容
内联并排

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']