diff --git a/arcface_converter/src/networks/__init__.py b/arcface_converter/src/models/__init__.py similarity index 100% rename from arcface_converter/src/networks/__init__.py rename to arcface_converter/src/models/__init__.py diff --git a/arcface_converter/src/networks/arcface_converter.py b/arcface_converter/src/models/arcface_converter.py similarity index 100% rename from arcface_converter/src/networks/arcface_converter.py rename to arcface_converter/src/models/arcface_converter.py diff --git a/arcface_converter/src/training.py b/arcface_converter/src/training.py index 07f8e4d..f843a52 100644 --- a/arcface_converter/src/training.py +++ b/arcface_converter/src/training.py @@ -10,7 +10,7 @@ from pytorch_lightning.tuner.tuning import Tuner from torch import Tensor from torch.utils.data import DataLoader, Dataset, TensorDataset, random_split -from .networks.arcface_converter import ArcFaceConverter +from .models.arcface_converter import ArcFaceConverter from .types import Batch, Loader CONFIG = configparser.ConfigParser() diff --git a/face_swapper/src/losses/__init__.py b/face_swapper/src/models/__init__.py similarity index 100% rename from face_swapper/src/losses/__init__.py rename to face_swapper/src/models/__init__.py