Update lama/saicinpainting/training/trainers/base.py

lama/saicinpainting/training/trainers/base.py

@@ -9,13 +9,13 @@ import torch.nn as nn
 import torch.nn.functional as F
 from torch.utils.data import DistributedSampler
 
-from saicinpainting.evaluation import make_evaluator
-from saicinpainting.training.data.datasets import make_default_train_dataloader, make_default_val_dataloader
-from saicinpainting.training.losses.adversarial import make_discrim_loss
-from saicinpainting.training.losses.perceptual import PerceptualLoss, ResNetPL
-from saicinpainting.training.modules import make_generator, make_discriminator
-from saicinpainting.training.visualizers import make_visualizer
-from saicinpainting.utils import add_prefix_to_keys, average_dicts, set_requires_grad, flatten_dict, \
+from lama.saicinpainting.evaluation import make_evaluator
+from lama.saicinpainting.training.data.datasets import make_default_train_dataloader, make_default_val_dataloader
+from lama.saicinpainting.training.losses.adversarial import make_discrim_loss
+from lama.saicinpainting.training.losses.perceptual import PerceptualLoss, ResNetPL
+from lama.saicinpainting.training.modules import make_generator, make_discriminator
+from lama.saicinpainting.training.visualizers import make_visualizer
+from lama.saicinpainting.utils import add_prefix_to_keys, average_dicts, set_requires_grad, flatten_dict, \
     get_has_ddp_rank
 
 LOGGER = logging.getLogger(__name__)