Commit graph

365 commits

Author SHA1 Message Date
iperov
bd0b408a9b fix 2021-09-06 16:11:23 +04:00
iperov
335da71db6 saehd: remove sample processor count limitation 2021-09-06 14:16:19 +04:00
iperov
70c9ee9463 change face_style_power 2021-09-06 14:11:08 +04:00
iperov
d301a4799f _ 2021-09-06 14:07:20 +04:00
iperov
b1990d421a AMP: don't train inters with random_warp-off 2021-09-03 13:50:31 +04:00
iperov
3fe8ce86b1 AMP: enable inter_src+inter_dst training 2021-08-30 18:52:29 +04:00
iperov
f346e35e7f fix 2021-08-30 12:57:19 +04:00
iperov
6e094d873d SAEHD: removed 'dst_denoise' option. Added -t arhi option. 2021-08-29 11:48:55 +04:00
iperov
01f1a084b4 AMP: remove dst_denoise option 2021-08-29 11:46:51 +04:00
iperov
e53be5e22d fix 2021-08-24 13:18:51 +04:00
iperov
91187ecb95 AMP: some arhi change results training stabilization. New options blur_out_mask + rtm_dst_denoise. Sample processors count are no more limited to 8, thus if you have AMD processor with 16+ cores, increase paging file size. 2021-08-24 12:58:23 +04:00
iperov
8e63666390 SAEHD: add 'Blur out mask' and 'Denoise DST faceset' options. 2021-08-20 17:07:44 +04:00
iperov
c5584fbda0 XSeg: changed pretrain mode 2021-08-12 17:01:38 +04:00
iperov
55b947eab5 XSeg: added pretrain option. 2021-07-30 17:24:21 +04:00
iperov
83b1412da7 Export AMP/SAEHD: added "Export quantized" option 2021-07-30 17:13:46 +04:00
iperov
d26d579836 AMP change help of ct mode. Default inter dims now 1024 2021-07-29 23:14:30 +04:00
iperov
9e1bc5a153 AMP: return lr_dropout option 2021-07-28 11:00:19 +04:00
iperov
4be135af60 AMP: last high loss samples behaviour - same as SAEHD 2021-07-28 08:58:32 +04:00
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