Ross Wightman d2e3c09ce1

Update version.py

2023-08-23 22:51:56 -07:00

..

data

…

layers

…

loss

…

models

Fix first_conv for mobileone and 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

MobileOne and FastViT weights on HF hub, more code cleanup and tweaks, features_only working. Add reparam flag to validate and benchmark, support reparm of all models with fuse(), reparameterize() or switch_to_deploy() methods on modules

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

__init__.py

…

version.py

Update version.py

2023-08-23 22:51:56 -07:00