From 9ff30a026882b2fd04ec131b14df49f67e1ad375 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 7 Mar 2025 14:07:00 +0100 Subject: [PATCH] Use getint --- face_swapper/src/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py index 7219ac5..ebf7270 100644 --- a/face_swapper/src/dataset.py +++ b/face_swapper/src/dataset.py @@ -16,7 +16,7 @@ from .types import Batch, BatchMode, WarpTemplate class DynamicDataset(Dataset[Tensor]): def __init__(self, config_parser : ConfigParser) -> None: self.config_file_pattern = config_parser.get('training.dataset', 'file_pattern') - self.config_transform_size = config_parser.get('training.dataset', 'transform_size') + self.config_transform_size = config_parser.getint('training.dataset', 'transform_size') self.config_batch_mode = cast(BatchMode, config_parser.get('training.dataset', 'batch_mode')) self.config_batch_ratio = config_parser.getfloat('training.dataset', 'batch_ratio') self.config_parser = config_parser