next you need to make saehd model to train with
if you already have model you can use export_saehd_config.bat (you need to have adabelief enabled)
open the saehd_config.yaml (with text editor like notepad)
you should see list like this
general_params:
name: model
model_params:
archi: liae-ud
resolution: 128
ae_dims: 256
e_dims: 64
d_dims: 64
d_mask_dims: 22
face_type: wf
data_loading:
use_cache: true
image_type: jpg
image_quality: 1
force_recache: false
folder_suffix: .face_cache_
training_properties:
mp_mode: full_t
pin_samples: true
autobackup_hour: 3
maximum_n_backups: 24
batch_size: 8
batch_dst_mixing: 0.0
masked_training: true
eye_mouth_priority: false
blur_out_mask: true
uniform_yaw: true
lr: 5.0e-05
lr_dropout: false
cpu_cap: 6
clipgrad: false
keep_graph: false
loss: ssim
weight_decay: 0.0005
freeze_encoder: false
snapshot_interval: 5000
augmentations:
random_warp: true
random_noise: false
random_src_flip: false
random_dst_flip: true
color_augmentation: true
scale_range: 0.15
rotation_range: 2
tx_range: 0.05
ty_range: 0.05
gan:
gan_training: false
gan_power: 0.01
gan_dims: 16
gan_patch_size: 64
gan_weight_decay: 0.001
gan_smoothing: 0.3
gan_noise: 0.01
you should change atleast these thinks (idk what most of the thinks do cause they are new and I haven't testing yet)
you should check this
You must be registered for see links
for optimal res and dims for your gpu (only vram should matter)
general_params:
name: model (put your model name can be anything!)
model_params:
archi: liae-ud (you can try with diffrent model arc (i use liae-ud))
resolution: 128 (look what other people use with same ammount of vram)
ae_dims: 256 (should be same with res or higer)
e_dims: 64 (you should again check what other people use with same ammount of vram)
d_dims: 64 (this number has to be same with e_dims or model corrupts)
d_mask_dims: 22 (idk about this i use 28)
face_type: wf (just use wf)
data_loading:
use_cache: true
image_type: jpg
image_quality: 1
force_recache: false
folder_suffix: .face_cache_
training_properties:
mp_mode: full_t (change this to bfloat it should be faster)
pin_samples: true
autobackup_hour: 3
maximum_n_backups: 24
batch_size: 8 (this should be 8 but you can use 6-10 but other values are considering bad)
batch_dst_mixing: 0.0
masked_training: true
eye_mouth_priority: false (use if eyes look weird after long amount of training)
blur_out_mask: true ()
uniform_yaw: true
lr: 5.0e-05
lr_dropout: false (need to enable this on some point but not in start)
cpu_cap: 6 (change this so it matches with the amount of you cores in your cpu)
clipgrad: false (change to true)
keep_graph: false
loss: ssim
weight_decay: 0.0005
freeze_encoder: false
snapshot_interval: 5000
augmentations:
random_warp: true (should be true at the start but need dissable at the end of training)
random_noise: false
random_src_flip: false
random_dst_flip: true
color_augmentation: true
scale_range: 0.15
rotation_range: 2
tx_range: 0.05
ty_range: 0.05
gan:
gan_training: false (you can enable this at the end of training)
gan_power: 0.01
gan_dims: 16
gan_patch_size: 64
gan_weight_decay: 0.001
gan_smoothing: 0.3
gan_noise: 0.01