fix 'basicsr.losses.losses' to 'basicsr.losses.gan_loss'

#3
by jone - opened
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