01f1a084b4
AMP: remove dst_denoise option
2021-08-29 11:46:51 +04:00
e53be5e22d
fix
2021-08-24 13:18:51 +04:00
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
8e63666390
SAEHD: add 'Blur out mask' and 'Denoise DST faceset' options.
2021-08-20 17:07:44 +04:00
c5584fbda0
XSeg: changed pretrain mode
2021-08-12 17:01:38 +04:00
55b947eab5
XSeg: added pretrain option.
2021-07-30 17:24:21 +04:00
83b1412da7
Export AMP/SAEHD: added "Export quantized" option
2021-07-30 17:13:46 +04:00
d26d579836
AMP change help of ct mode. Default inter dims now 1024
2021-07-29 23:14:30 +04:00
9e1bc5a153
AMP: return lr_dropout option
2021-07-28 11:00:19 +04:00
4be135af60
AMP: last high loss samples behaviour - same as SAEHD
2021-07-28 08:58:32 +04:00
336e0ca944
fix
2021-07-17 22:04:20 +04:00
0748b8d043
AMP: removed eyes mouth prio option(default enabled), removed masked training option(default enabled).
2021-07-17 21:58:57 +04:00
959a3530f8
AMP: default morph factor is now 0.5. Preview shows with 1.0
2021-07-17 10:27:07 +04:00
da8f33ee85
removing fp16 for amp
2021-07-15 13:23:33 +04:00
c9e0ba1779
removing fp16 for sae
2021-07-15 13:22:32 +04:00
7b8991a3cc
fix help
2021-07-15 00:50:34 +04:00
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
ee1bc83a14
AMP, SAEHD: reverted GAN to december version.
2021-07-07 21:45:43 +04:00
446b95942f
fix
2021-07-01 15:08:11 +04:00
0444a9a565
default morph factor is now 0.1
2021-07-01 14:29:11 +04:00
075422e0c5
amp: fixes, removed pretrain option
2021-07-01 14:25:54 +04:00
63c794b3d0
rename dump_dflive to export_dfm
2021-06-28 18:26:54 +04:00
9d6b6feb1f
_
2021-06-28 14:27:09 +04:00
90a74efd89
fix
2021-06-26 23:06:30 +04:00
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
5783191849
AMP: code refactoring, fix preview history
...
added dumpdflive command
2021-06-26 10:44:41 +04:00
62f1d57871
AMP: fix a merge bug where morph_factor was always = 1.0
2021-06-14 17:58:01 +04:00
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
34b62862e0
fix help msg
2021-06-01 09:36:53 +04:00
e52b53f87c
AMP fix
2021-05-30 09:24:23 +04:00
11a7993238
fix
2021-05-29 22:23:07 +04:00
6f86c68e65
add AMP model
2021-05-29 18:57:35 +04:00
fdb143ff47
added AMD/Intel cards support via DirectX12 ( DirectML backend )
2021-04-22 18:19:15 +04:00
93fe480eca
revert
2021-04-19 13:35:14 +04:00
af0b3904fc
XSeg trainer: additional dssim loss.
2021-04-15 21:43:47 +04:00
8d46cd94fd
upd pretrain option help
2021-03-23 16:07:50 +04:00
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
b333fcea4b
dump_ckpt
2021-03-23 15:00:24 +04:00
11add4cd4f
update GAN model
2021-01-03 19:25:39 +04:00
54fc3162ed
new gan
2021-01-03 01:08:20 +04:00
ad5733c5bb
upd PatchDiscriminator
2020-12-31 17:38:41 +04:00
ae9e16b4a5
SAEHD: new GAN
2020-12-30 14:33:14 +04:00
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
dd037d2dea
SAEHD: lr_dropout now can be used with AdaBelief
2020-12-18 12:42:10 +04:00
40d18896b7
SAEHD: change help
2020-12-16 18:25:25 +04:00
da058cf17f
remove lrcos from adabelief
2020-12-11 20:34:14 +04:00
26ed582aff
fix xseg
2020-12-11 20:25:30 +04:00
1ee7798fb9
.
2020-12-11 17:56:39 +04:00
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
254a7cf5cf
fix xseg training
2020-12-11 15:47:11 +04:00
iperov