mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-06-25 07:59:55 +02:00
harisreedhar
72 lines
1.6 KiB
INI
72 lines
1.6 KiB
INI
[preparing.dataset]
|
|
dataset_path = /assets/VGGface2_None_norm_512_true_bygfpgan
|
|
|
|
[preparing.dataloader]
|
|
same_person_probability = 0.2
|
|
|
|
[preparing.augmentation]
|
|
expression = false
|
|
|
|
[training.loader]
|
|
batch_size = 4
|
|
num_workers = 8
|
|
|
|
[training.generator]
|
|
num_blocks = 2
|
|
id_channels = 512
|
|
learning_rate = 0.0004
|
|
|
|
[training.discriminator]
|
|
input_channels = 3
|
|
num_filters = 64
|
|
num_layers = 5
|
|
num_discriminators = 3
|
|
learning_rate = 0.0004
|
|
disable = false
|
|
|
|
[auxiliary_models.paths]
|
|
arcface_path = /assets/pretrained_models/arcface_w600k_r50.pt
|
|
landmarker_path = /assets/pretrained_models/landmark_203.pt
|
|
motion_extractor_path = /assets/pretrained_models/liveportrait_motion_extractor.pth
|
|
feature_extractor_path = /assets/pretrained_models/liveportrait_feature_extractor.pth
|
|
warping_network_path = /assets/pretrained_models/liveportrait_warping_model.pth
|
|
spade_generator_path = /assets/pretrained_models/liveportrait_spade_generator.pth
|
|
|
|
[training.losses]
|
|
weight_adversarial = 1
|
|
weight_identity = 20
|
|
weight_attribute = 10
|
|
weight_reconstruction = 10
|
|
weight_tsr = 100
|
|
weight_eye_gaze = 5
|
|
weight_eye_open = 5
|
|
weight_lip_open = 5
|
|
|
|
[training.schedulers]
|
|
step = 5000
|
|
gamma = 0.2
|
|
|
|
[training.trainer]
|
|
max_epochs = 50
|
|
disable_discriminator = false
|
|
|
|
[training.output]
|
|
checkpoint_path = checkpoints/last.ckpt
|
|
directory_path = checkpoints
|
|
file_pattern = 'checkpoint-{epoch}-{step}-{l_G:.4f}-{l_D:.4f}'
|
|
preview_frequency = 250
|
|
validation_frequency = 1000
|
|
|
|
[training.validation]
|
|
sources = assets/test/front/sources
|
|
targets = assets/test/front/targets
|
|
|
|
[exporting]
|
|
directory_path =
|
|
source_path =
|
|
target_path =
|
|
opset_version =
|
|
|
|
[execution]
|
|
providers =
|