diff --git a/hyperswap/README.md b/hyperswap/README.md index ab4b173..538f0b2 100644 --- a/hyperswap/README.md +++ b/hyperswap/README.md @@ -29,9 +29,9 @@ This `config.ini` utilizes the VGGFace2 dataset to train the HyperSwap model. [training.dataset] file_pattern = .datasets/vggface2/**/*.jpg convert_template = vggfacehq_512_to_arcface_128 +multiplier = 1 transform_size = 256 usage_mode = both -usage_ratio = 1 batch_mode = same batch_ratio = 0.2 ``` diff --git a/hyperswap/config.ini b/hyperswap/config.ini index 965860a..1f979c7 100644 --- a/hyperswap/config.ini +++ b/hyperswap/config.ini @@ -1,9 +1,9 @@ [training.dataset] file_pattern = convert_template = +multiplier transform_size = usage_mode = -usage_ratio = batch_mode = batch_ratio = diff --git a/hyperswap/src/training.py b/hyperswap/src/training.py index 41a952c..0e57faf 100644 --- a/hyperswap/src/training.py +++ b/hyperswap/src/training.py @@ -208,7 +208,7 @@ def prepare_datasets(config_parser : ConfigParser) -> List[Dataset[Tensor]]: for config_section in config_parser.sections(): if config_section.startswith('training.dataset'): - config_usage_ratio = config_parser.getint(config_section, 'usage_ratio') + config_multiplier = config_parser.getint(config_section, 'multiplier') __config_parser__ = deepcopy(config_parser) __config_parser__.remove_section(config_section) __config_parser__.add_section('training.dataset.current') @@ -217,7 +217,7 @@ def prepare_datasets(config_parser : ConfigParser) -> List[Dataset[Tensor]]: __config_parser__.set('training.dataset.current', key, value) dynamic_dataset = DynamicDataset(__config_parser__) - datasets.extend([ dynamic_dataset ] * config_usage_ratio) + datasets.extend([ dynamic_dataset ] * config_multiplier) return datasets