snicolau commited on
Commit
f537759
β€’
1 Parent(s): 31cd0b1

Update lama/saicinpainting/evaluation/losses/base_loss.py

Browse files
lama/saicinpainting/evaluation/losses/base_loss.py CHANGED
@@ -10,7 +10,7 @@ import torch.nn.functional as F
10
  from joblib import Parallel, delayed
11
  from scipy import linalg
12
 
13
- from models.ade20k import SegmentationModule, NUM_CLASS, segm_options
14
  from .fid.inception import InceptionV3
15
  from .lpips import PerceptualLoss
16
  from .ssim import SSIM
 
10
  from joblib import Parallel, delayed
11
  from scipy import linalg
12
 
13
+ from lama.models.ade20k import SegmentationModule, NUM_CLASS, segm_options
14
  from .fid.inception import InceptionV3
15
  from .lpips import PerceptualLoss
16
  from .ssim import SSIM