原丹训练了105万次没开gan,我开gan又训练大约11万次,带佟丽娅素材去重四千多张,不满意可以继续训练,8G以上显卡都可以,AMP模型特点光影表现比liae好,源相似度差一些。
==================== Model Summary ====================
== ==
== Model name: AMPSRCSRC_AMP ==
== ==
== Current iteration: 6160017 ==
== ==
==—————— Model Options ——————==
== ==
== resolution: 224 ==
== face_type: wf ==
== models_opt_on_gpu: True ==
== ae_dims: 512 ==
== inter_dims: 1024 ==
== e_dims: 64 ==
== d_dims: 64 ==
== d_mask_dims: 22 ==
== morph_factor: 0.5 ==
== uniform_yaw: False ==
== blur_out_mask: True ==
== lr_dropout: y ==
== random_warp: False ==
== ct_mode: rct ==
== clipgrad: False ==
== autobackup_hour: 0 ==
== write_preview_history: False ==
== target_iter: 0 ==
== random_src_flip: False ==
== random_dst_flip: False ==
== batch_size: 8 ==
== gan_power: 0.04 ==
== gan_patch_size: 28 ==
== gan_dims: 32 ==
== retraining_samples: False ==
== preview_mf: 1 ==
== masked_training: True ==
== eyes_prio: False ==
== mouth_prio: False ==
== loss_function: SSIM ==
== adabelief: True ==
== random_hsv_power: 0.0 ==
== random_downsample: False ==
== random_noise: False ==
== random_blur: False ==
== random_jpeg: False ==
== random_shadow: none ==
== background_power: 0.0 ==
== random_color: False ==
== use_fp16: False ==
== cpu_cap: 8 ==
== preview_samples: 8 ==
== force_full_preview: False ==
== lr: 5e-05 ==
== session_name: ==
== maximum_n_backups: 24 ==
== gan_smoothing: 0.1 ==
== gan_noise: 0.0 ==
== ==
==——————- Running On ——————–==
== ==
== Device index: 0 ==
== Name: NVIDIA GeForce RTX 2080 Ti ==
== VRAM: 8.73GB ==
== ==
=======================================================