pytorch-image-models/timm
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