iperov 55b947eab5 XSeg: added pretrain option.

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

..

archis

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

initializers

global refactoring and fixes,

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

layers

remove ConvolutionAwareInitializer from layers, because lr-dropout does the same thing.

2021-07-15 00:50:11 +04:00

models

XSeg: added pretrain option.

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

ops

AMP fix

2021-05-30 09:24:23 +04:00

optimizers

fix for clipgrad

2021-07-15 00:24:38 +04:00

__init__.py

…

device.py

fix for colab

2021-04-22 21:12:42 +04:00

nn.py

added AMD/Intel cards support via DirectX12 ( DirectML backend )

2021-04-22 18:19:15 +04:00