Revert "fix basicsr losses import (TencentARC#210)"

This reverts commit 7d657f2
AdamBear · Aug 16, 2022 · 8ef599d · 8ef599d
1 parent 697d93f
commit 8ef599d
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.gan_loss import r1_penalty
+from basicsr.losses.losses 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