iperov 55b947eab5 XSeg: added pretrain option.

2021-07-30 17:24:21 +04:00

..

__init__.py

New script:

2020-03-30 14:00:40 +04:00

CodeDiscriminator.py

global refactoring and fixes,

2020-03-13 08:09:00 +04:00

ModelBase.py

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

PatchDiscriminator.py

AMP, SAEHD: added option use_fp16: Increases training/inference speed, reduces model size. Model may crash. Enable it after 1-5k iters.

2021-07-14 23:03:54 +04:00

XSeg.py

XSeg: added pretrain option.

2021-07-30 17:24:21 +04:00