fix basicsr losses import (TencentARC#210)

PhotoDay · Jul 13, 2022 · 5c03750 · 5c03750
1 parent 3fd7a2f
commit 5c03750
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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