diff --git a/ppgan/models/edvr_model.py b/ppgan/models/edvr_model.py index 86f269aa1e0592d1871ed9a128ce7b6d4faedd86..baca93b36fc125c508ba10ccfe465df5cc411265 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']