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	DFL-2.0 initial branch commit	2020-01-21 18:43:39 +04:00
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