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

..

Model_Quick96

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

Model_SAEHD

SAEHD: added new option Use AdaBelief optimizer?

2020-12-11 15:48:27 +04:00

Model_XSeg

fix xseg training

2020-12-11 15:47:11 +04:00

__init__.py

DFL-2.0 initial branch commit

2020-01-21 18:43:39 +04:00

ModelBase.py

options_show_override{} for model summary in pretrain mode

2020-07-27 16:37:13 +04:00