Ross Wightman c6e4b7895a Swin V2 CR impl refactor. ...

* reformat and change some naming so closer to existing timm vision transformers
* remove typing that wasn't adding clarity (or causing torchscript issues)
* support non-square windows
* auto window size adjust from image size
* post-norm + main-branch no

2022-02-23 17:28:52 -08:00

..

data

Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.

2022-01-14 16:28:27 -08:00

loss

Update binary cross ent impl to use thresholding as an option (convert soft targets from mixup/cutmix to 0, 1)

2021-09-23 15:45:39 -07:00

models

Swin V2 CR impl refactor.

2022-02-23 17:28:52 -08:00

optim

fix lars

2022-01-19 17:49:43 +08:00

scheduler

Fix k_decay default arg != 1.0 in poly scheduler

2021-11-22 09:37:36 -08:00

utils

Add --fuser arg to train/validate/benchmark scripts to select jit fuser type

2022-01-17 13:54:25 -08: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

version 0.5.5

2022-01-20 21:18:30 -08:00