Ross Wightman cac0a4570a More test fixes, pool size for 256x256 maxvit models

2022-08-23 13:38:26 -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

More test fixes, pool size for 256x256 maxvit models

2022-08-23 13:38:26 -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