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

..

__init__.py

add AMP model

2021-05-29 18:57:35 +04:00

Model.py

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