CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-04-19 15:56:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

Rename to dataset multiplier

Browse Source
This commit is contained in:
henryruhs
2025-06-05 16:34:03 +02:00
parent 47bebb02d7
commit 143b594ee6
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hyperswap/README.md
+1 -1
View File
@@ -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
```
hyperswap/config.ini
+1 -1
View File
@@ -1,9 +1,9 @@
[training.dataset]
file_pattern =
convert_template =
multiplier
transform_size =
usage_mode =
usage_ratio =
batch_mode =
batch_ratio =
hyperswap/src/training.py
+2 -2
View File
@@ -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