Final rename for everything

2026-06-25 07:59:55 +02:00 · 2025-04-24 12:42:53 +02:00
parent 03011200e4
commit 810df0f540
41 changed files with 44 additions and 44 deletions
@@ -0,0 +1,24 @@
+from typing import Any, Dict, Literal, Tuple, TypeAlias
+
+from torch import Tensor
+from torch.nn import Module
+
+Batch : TypeAlias = Tuple[Tensor, Tensor]
+BatchMode = Literal['equal', 'same', 'different']
+
+Feature : TypeAlias = Tensor
+Embedding : TypeAlias = Tensor
+Mask : TypeAlias = Tensor
+Loss : TypeAlias = Tensor
+
+Padding : TypeAlias = Tuple[int, int, int, int]
+
+GeneratorModule : TypeAlias = Module
+EmbedderModule : TypeAlias = Module
+GazerModule : TypeAlias = Module
+FaceMaskerModule : TypeAlias = Module
+
+OptimizerSet : TypeAlias = Any
+
+WarpTemplate = Literal['arcface_128_v2_to_arcface_112_v2', 'ffhq_512_to_arcface_128_v2', 'vggfacehq_256_to_arcface_128_v2']
+WarpTemplateSet : TypeAlias = Dict[WarpTemplate, Tensor]