From 5da06e50b36e684420eaa8ddf82b08987c5fb9fc Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Mon, 31 May 2021 07:43:29 +0000
Subject: [PATCH] fix edvr train on rocm

---
 ppgan/models/edvr_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppgan/models/edvr_model.py b/ppgan/models/edvr_model.py
index 86f269a..baca93b 100644
--- 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']
 
-- 
GitLab