From 7d657f26b66c3e01a969d1598b41ce65c9883c55 Mon Sep 17 00:00:00 2001
From: AJ <46843456+amckenna41@users.noreply.github.com>
Date: Wed, 13 Jul 2022 03:01:06 +0100
Subject: [PATCH] fix basicsr losses import (#210)

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

diff --git a/gfpgan/models/gfpgan_model.py b/gfpgan/models/gfpgan_model.py
index 684fc601..b5fb8c95 100644
--- a/gfpgan/models/gfpgan_model.py
+++ b/gfpgan/models/gfpgan_model.py
@@ -3,7 +3,7 @@
 import torch
 from basicsr.archs import build_network
 from basicsr.losses import build_loss
-from basicsr.losses.losses import r1_penalty
+from basicsr.losses.gan_loss import r1_penalty
 from basicsr.metrics import calculate_metric
 from basicsr.models.base_model import BaseModel
 from basicsr.utils import get_root_logger, imwrite, tensor2img