akhaliq HF staff jone commited on
Commit
4316d1d
1 Parent(s): b1e629c

fix 'basicsr.losses.losses' to 'basicsr.losses.gan_loss' (#3)

Browse files

- fix 'basicsr.losses.losses' to 'basicsr.losses.gan_loss' (90e8102e1622a944ef20823a66ae70589210ef48)


Co-authored-by: king <jone@users.noreply.huggingface.co>

Files changed (1) hide show
  1. gfpgan/models/gfpgan_model.py +1 -1
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.losses 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
 
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