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
|