Spaces:
Runtime error
Runtime error
fix 'basicsr.losses.losses' to 'basicsr.losses.gan_loss'
#3
by
jone
- opened
gfpgan/models/gfpgan_model.py
CHANGED
@@ -3,7 +3,7 @@ import os.path as osp
|
|
3 |
import torch
|
4 |
from basicsr.archs import build_network
|
5 |
from basicsr.losses import build_loss
|
6 |
-
from basicsr.losses.
|
7 |
from basicsr.metrics import calculate_metric
|
8 |
from basicsr.models.base_model import BaseModel
|
9 |
from basicsr.utils import get_root_logger, imwrite, tensor2img
|
|
|
3 |
import torch
|
4 |
from basicsr.archs import build_network
|
5 |
from basicsr.losses import build_loss
|
6 |
+
from basicsr.losses.gan_loss import r1_penalty
|
7 |
from basicsr.metrics import calculate_metric
|
8 |
from basicsr.models.base_model import BaseModel
|
9 |
from basicsr.utils import get_root_logger, imwrite, tensor2img
|