iperov
336e0ca944
fix
2021-07-17 22:04:20 +04:00
iperov
0748b8d043
AMP: removed eyes mouth prio option(default enabled), removed masked training option(default enabled).
2021-07-17 21:58:57 +04:00
iperov
959a3530f8
AMP: default morph factor is now 0.5. Preview shows with 1.0
2021-07-17 10:27:07 +04:00
iperov
da8f33ee85
removing fp16 for amp
2021-07-15 13:23:33 +04:00
iperov
c9e0ba1779
removing fp16 for sae
2021-07-15 13:22:32 +04:00
iperov
7b8991a3cc
fix help
2021-07-15 00:50:34 +04:00
iperov
35877dbfd7
AMP, SAEHD: added option use_fp16: Increases training/inference speed, reduces model size. Model may crash. Enable it after 1-5k iters.
...
AMP: removed lr_dropout, now it is enabled by default;
2021-07-14 23:03:54 +04:00
iperov
ee1bc83a14
AMP, SAEHD: reverted GAN to december version.
2021-07-07 21:45:43 +04:00
iperov
446b95942f
fix
2021-07-01 15:08:11 +04:00
iperov
0444a9a565
default morph factor is now 0.1
2021-07-01 14:29:11 +04:00
iperov
075422e0c5
amp: fixes, removed pretrain option
2021-07-01 14:25:54 +04:00
iperov
63c794b3d0
rename dump_dflive to export_dfm
2021-06-28 18:26:54 +04:00
iperov
9d6b6feb1f
_
2021-06-28 14:27:09 +04:00
iperov
90a74efd89
fix
2021-06-26 23:06:30 +04:00
iperov
3c6bbe22b9
AMP: added Inter dimensions param. The model is not changed. Should be equal or more than AutoEncoder dimensions. More dims are better, but require more VRAM. You can fine-tune model size to fit your GPU.
2021-06-26 10:55:33 +04:00
iperov
5783191849
AMP: code refactoring, fix preview history
...
added dumpdflive command
2021-06-26 10:44:41 +04:00
iperov
62f1d57871
AMP: fix a merge bug where morph_factor was always = 1.0
2021-06-14 17:58:01 +04:00
iperov
5fac5ee4f3
AMP model: added ‘morph_factor’ option. [0.1 .. 0.5]
...
The smaller the value, the more src-like facial expressions will appear.
The larger the value, the less space there is to train a large dst faceset in the neural network.
Typical fine value is 0.33
AMP model: added ‘pretrain’ mode as in SAEHD
Windows build: Default pretrain dataset is updated with applied Generic XSeg mask
2021-06-01 09:37:55 +04:00
iperov
34b62862e0
fix help msg
2021-06-01 09:36:53 +04:00
iperov
e52b53f87c
AMP fix
2021-05-30 09:24:23 +04:00
iperov
11a7993238
fix
2021-05-29 22:23:07 +04:00
iperov
6f86c68e65
add AMP model
2021-05-29 18:57:35 +04:00
iperov
fdb143ff47
added AMD/Intel cards support via DirectX12 ( DirectML backend )
2021-04-22 18:19:15 +04:00
iperov
93fe480eca
revert
2021-04-19 13:35:14 +04:00
iperov
af0b3904fc
XSeg trainer: additional dssim loss.
2021-04-15 21:43:47 +04:00
iperov
8d46cd94fd
upd pretrain option help
2021-03-23 16:07:50 +04:00
iperov
e47b602ec8
SAEHD: random flip replaced with random SRC flip(default False) and random DST flip(default True)
2021-03-23 15:13:10 +04:00
iperov
b333fcea4b
dump_ckpt
2021-03-23 15:00:24 +04:00
iperov
11add4cd4f
update GAN model
2021-01-03 19:25:39 +04:00
iperov
54fc3162ed
new gan
2021-01-03 01:08:20 +04:00
iperov
ad5733c5bb
upd PatchDiscriminator
2020-12-31 17:38:41 +04:00
iperov
ae9e16b4a5
SAEHD: new GAN
2020-12-30 14:33:14 +04:00
iperov
db83a21244
Eyes priority is replaced with Eyes and mouth priority,
...
Helps to fix eye problems during training like "alien eyes" and wrong eyes direction.
Also makes the detail of the teeth higher.
New default values with new model:
Archi : ‘liae-ud’
AdaBelief : enabled
2020-12-20 09:45:22 +04:00
iperov
dd037d2dea
SAEHD: lr_dropout now can be used with AdaBelief
2020-12-18 12:42:10 +04:00
iperov
40d18896b7
SAEHD: change help
2020-12-16 18:25:25 +04:00
iperov
da058cf17f
remove lrcos from adabelief
2020-12-11 20:34:14 +04:00
iperov
26ed582aff
fix xseg
2020-12-11 20:25:30 +04:00
iperov
1ee7798fb9
.
2020-12-11 17:56:39 +04:00
iperov
704b5dc072
SAEHD: added new option Use AdaBelief optimizer?
...
Experimental AdaBelief optimizer. It requires more VRAM, but the accuracy of the model is higher, and lr_dropout is not needed.
2020-12-11 15:48:27 +04:00
iperov
254a7cf5cf
fix xseg training
2020-12-11 15:47:11 +04:00
iperov
35945b257c
fix dimensions error
2020-12-11 14:34:37 +04:00
iperov
b9c9e7cffd
fix depth_to_space for tf2.4.0. Removing compute_output_shape in leras, because it uses CPU device, which does not support all ops.
2020-12-11 11:28:33 +04:00
Colombo
6134e57762
options_show_override{} for model summary in pretrain mode
2020-07-27 16:37:13 +04:00
Colombo
1435dd3dd1
SAEHD: now random_warp is disabled for pretraining mode by default
2020-07-26 16:11:08 +04:00
Colombo
ce95af9068
SAEHD: write_preview_history now works faster
...
The frequency at which the preview is saved now depends on the resolution.
For example 64x64 – every 10 iters. 448x448 – every 70 iters.
2020-07-18 11:03:13 +04:00
Colombo
58670722dc
fix
2020-07-17 17:40:32 +04:00
Colombo
b9b97861e1
fix mask for style losses
2020-07-17 17:34:07 +04:00
Colombo
e8b04053e4
SAEHD:
...
Changed help for “Learning rate dropout” option:
When the face is trained enough, you can enable this option to get extra sharpness and reduce subpixel shake for less amount of iterations.
Enabled it before “disable random warp” and before GAN. n disabled. y enabled
cpu enabled on CPU. This allows not to use extra VRAM, sacrificing 20% time of iteration.
Changed help for GAN option:
Train the network in Generative Adversarial manner.
Forces the neural network to learn small details of the face.
Enable it only when the face is trained enough and don't disable.
Typical value is 0.1
improved GAN. Now it produces better skin detail, less patterned aggressive artifacts, works faster.
https://i.imgur.com/Nbh3mw1.png
2020-07-16 22:19:17 +04:00
Colombo
44e6970dc5
changed help for gan_power
2020-06-30 12:37:12 +04:00
Colombo
4e998aff93
.
2020-06-24 10:06:31 +04:00