Ross Wightman 01cb46a9a5 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

..

data

Fix tf.data options setting for newer TF versions

2021-05-15 22:56:12 -07:00

loss

asymmetric_loss

2020-10-16 17:12:28 +03:00

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

Fix adabelief imports, remove prints, preserve memory format is the default arg for zeros_like

2021-04-12 08:25:31 -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

Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models

2021-04-12 11:54:22 -07:00

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