CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix typing

Browse Source
This commit is contained in:
henryruhs
2025-02-21 09:17:59 +01:00
parent 15ee6fa763
commit 4078681031
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/training.py
+1 -1
View File
@@ -83,7 +83,7 @@ def create_loaders(dataset : Dataset[Tensor]) -> Tuple[DataLoader[Tensor], DataL
def split_dataset(dataset : Dataset[Tensor]) -> Tuple[Dataset[Tensor], Dataset[Tensor]]:
loader_split_ratio = CONFIG.getfloat('training.loader', 'split_ratio')
dataset_size = len(dataset)
dataset_size = len(dataset) # type:ignore[arg-type]
training_size = dataset_size * loader_split_ratio
validation_size = dataset_size - training_size
training_dataset, validate_dataset = random_split(dataset, [ training_size, validation_size ])
embedding_converter/src/types.py
+1 -1
View File
@@ -7,4 +7,4 @@ Batch : TypeAlias = Tensor
Embedding : TypeAlias = Tensor
VisionFrame : TypeAlias = NDArray[Any]
OptimizerConfig : TypeAlias = Dict[str, Any]
OptimizerConfig : TypeAlias = Any