Ross Wightman c7a20cec13 Begin adding FastViT

2023-08-23 22:50:37 -07:00

..

data

Replace deprecated NumPy aliases of builtin types

2023-07-03 22:24:25 +03:00

layers

Add more advanced interpolation method from BEiT and support non-square window & image size adaptation for

2023-08-08 16:41:16 -07:00

loss

…

models

Begin adding FastViT

2023-08-23 22:50:37 -07:00

optim

import lion in __init__.py

2023-07-27 09:26:57 +08:00

scheduler

Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168

2022-10-07 10:43:04 -07:00

utils

clip gradients with update

2023-04-19 23:36:20 -07:00

__init__.py

…

version.py

Version 0.9.5 prep for release. README update

2023-08-03 15:16:46 -07:00