Ross Wightman 1042b8a146 Add non fused LAMB optimizer option		2021-08-09 13:13:43 -07:00
..
data	Fix tf.data options setting for newer TF versions	2021-05-15 22:56:12 -07:00
loss	…
models	Add gc_efficientnetv2_rw_t weights (global context instead of SE attn). Add TF XL weights even though the fine-tuned ones don't validate that well. Change default arg for GlobalContext to use scal (mul) mode.	2021-08-07 16:45:29 -07:00
optim	Add non fused LAMB optimizer option	2021-08-09 13:13:43 -07:00
scheduler	1. Added a simple multi step LR scheduler	2021-07-09 16:18:27 -04:00
utils	Merge pull request #550 from amaarora/wandb	2021-04-15 09:26:35 -07:00
__init__.py	…
version.py	Add efficientnetv2_rw_t defs w/ weights, and gc variant, as well as gcresnet26ts for experiments. Version 0.4.13	2021-07-09 16:46:52 -07:00