Ross Wightman 63417b438f TinyViT adjustments ...

* keep most of net in BCHW layout, performance appears same, can remove static resolution attribs and features easier to use
* add F.sdpa, decent gains in pt 2.1
* tweak crop pct based on eval

2023-09-01 11:05:56 -07:00

..

data

…

layers

add tinyvit

2023-09-01 11:05:56 -07:00

loss

…

models

TinyViT adjustments

2023-09-01 11:05:56 -07:00

optim

…

scheduler

…

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

Version 0.9.6 for release

2023-08-29 09:14:13 -07:00