DeepFaceLab/nnlib
iperov 37505d88e3 old SAE model will not work with this update.
Fixed bug when SAE can be collapsed during a time.

SAE: removed CA weights and encoder/decoder dims.

added new options:

Encoder dims per channel (21-85 ?:help skip:%d)
More encoder dims help to recognize more facial features, but require more VRAM. You can fine-tune model size to fit your GPU.

Decoder dims per channel (11-85 ?:help skip:%d)
More decoder dims help to get better details, but require more VRAM. You can fine-tune model size to fit your GPU.

Add residual blocks to decoder? (y/n, ?:help skip:n) :
These blocks help to get better details, but require more computing time.

Remove gray border? (y/n, ?:help skip:n) :
Removes gray border of predicted face, but requires more computing resources.
2019-03-24 15:35:02 +04:00
..
__init__.py removing trailing spaces 2019-03-19 23:53:27 +04:00
CAInitializer.py removing trailing spaces 2019-03-19 23:53:27 +04:00
device.py removing trailing spaces 2019-03-19 23:53:27 +04:00
nnlib.py old SAE model will not work with this update. 2019-03-24 15:35:02 +04:00
pynvml.py removing trailing spaces 2019-03-19 23:53:27 +04:00