Ross Wightman c207e02782 MOAR optimizer changes. Woo!

2021-08-18 22:20:35 -07:00

..

data

Add mmax config key to auto_augment for increasing upper bound of RandAugment magnitude beyond 10. Make AugMix uniform sampling default not override config setting.

2021-08-12 15:39:05 -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

MOAR optimizer changes. Woo!

2021-08-18 22:20:35 -07:00

scheduler

Update scheduler_factory.py

2021-08-14 22:53:17 +02: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