Ross Wightman a1cb25066e Add edgnext_small_rw weights trained with swin like recipe. Better than original 'small' but not the recent 'USI' distilled weights.

2022-07-07 22:02:57 -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 edgnext_small_rw weights trained with swin like recipe. Better than original 'small' but not the recent 'USI' distilled weights.

2022-07-07 22:02:57 -07:00

optim

…

scheduler

…

utils

Disable nvfuser fma / opt level overrides per #1244

2022-05-13 09:27:13 -07:00

__init__.py

…

version.py

Make dev version 0.6.2.dev0 for pypi pre

2022-05-15 14:29:57 -07:00