方曦 fabc4e5bcd Fixing tinyvit torchscript issue

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

..

data

…

layers

add tinyvit

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

loss

…

models

Fixing tinyvit torchscript issue

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