From b7e2d3ccd7c957ce75606fbbc4d6e05573aa97c0 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sat, 8 Feb 2025 02:54:29 +0100 Subject: [PATCH] It should be models and networks --- arcface_converter/src/{networks => models}/__init__.py | 0 arcface_converter/src/{networks => models}/arcface_converter.py | 0 arcface_converter/src/training.py | 2 +- face_swapper/src/{losses => models}/__init__.py | 0 4 files changed, 1 insertion(+), 1 deletion(-) rename arcface_converter/src/{networks => models}/__init__.py (100%) rename arcface_converter/src/{networks => models}/arcface_converter.py (100%) rename face_swapper/src/{losses => models}/__init__.py (100%) 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