Ross Wightman 8449ba210c Improve performance of HaloAttn, change default dim calc. Some cleanup / fixes for byoanet. Rename resnet26ts to tfs to distinguish (extra fc).

2021-08-26 21:56:44 -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

Improve performance of HaloAttn, change default dim calc. Some cleanup / fixes for byoanet. Rename resnet26ts to tfs to distinguish (extra fc).

2021-08-26 21:56:44 -07:00

optim

Add lamb/lars to optim init imports, remove stray comment

2021-08-18 22:55:02 -07:00

scheduler

Update scheduler_factory.py

2021-08-14 22:53:17 +02:00

utils

Tweak accuracy topk safety. Fix #807

2021-08-19 14:18:53 -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