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

2020-12-28 18:37:29 +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

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

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

upgrade to tf 2.4.0rc1

2020-11-13 17:00:07 +04:00