iperov 4f2efd7985 fix support for v1/v2

2021-01-01 17:59:57 +04:00

..

archis

fix dimensions error

2020-12-11 14:34:37 +04:00

initializers

global refactoring and fixes,

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

layers

leras: added AdaBelief optimizer

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

models

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

ops

leras ops: pixelnorm, total_variation_mse

2020-12-30 14:32:07 +04:00

optimizers

leras: added AdaBelief optimizer

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

__init__.py

DFL-2.0 initial branch commit

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

device.py

set CUDA_​CACHE_​MAXSIZE to 2gb before first cuda load

2020-12-28 18:37:29 +04:00

nn.py

fix support for v1/v2

2021-01-01 17:59:57 +04:00