Ross Wightman 7624389fc9 Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode.

2025-05-20 14:38:50 -07:00

..

data

Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode.

2025-05-20 14:38:50 -07:00

layers

Improve several typing issues for flex vit, can (almost) work with jit if we bash h,w key into an int or str

2025-04-14 11:01:56 -07:00

loss

…

models

Further pos embed tweaks, rejig model defs for testing

2025-04-28 09:15:11 -07:00

optim

Fix issue taking device from V before V exists

2025-01-31 12:52:47 -08:00

scheduler

…

utils

Fix typos

2025-01-19 13:39:40 -08:00

__init__.py

Use import alias

2025-01-22 10:27:17 -08:00

py.typed

…

version.py

Update version.py

2025-02-22 10:50:21 -08:00