Ross Wightman 43aa84e861 Add 'fast' layer norm that doesn't cast to float32, support APEX LN impl for slight speed gain, update norm and act factories, tweak SE for ability to disable bias (needed by GCVit)

2022-08-17 14:32:58 -07:00

..

data

Change set -> dict w/ None keys for dataset split synonym search, so always consistent if more than 1 exists. Fix #1224

2022-07-07 15:33:53 -07:00

loss

…

models

Add 'fast' layer norm that doesn't cast to float32, support APEX LN impl for slight speed gain, update norm and act factories, tweak SE for ability to disable bias (needed by GCVit)

2022-08-17 14:32:58 -07:00

optim

Fix bug introduced in non layer_decay weight_decay application. Remove debug print, fix arg desc.

2022-02-28 17:06:32 -08:00

scheduler

fix: change milestones to decay-milestones

2022-05-10 07:57:19 +09:00

utils

Add bulk_runner script and updates to benchmark.py and validate.py for better error handling in bulk runs (used for benchmark and validation result runs). Improved batch size decay stepping on retry...

2022-07-18 18:04:54 -07:00

__init__.py

Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.

2022-01-25 21:54:13 -08:00

version.py

Add timm ConvNeXt 'atto' weights, change test resolution for FB ConvNeXt 224x224 weights, add support for different dw kernel_size

2022-08-15 17:56:08 -07:00