Ross Wightman aa4d06a11c sbb vit weights on hub, testing

2024-05-10 17:15:01 -07:00

..

data

Add teddy-bear class back to first 1000 classes of imagenet22k_ms_synsets (index 851)

2024-04-09 09:33:08 -07:00

layers

Remove JIT activations, take jit out of ME activations. Remove other instances of torch.jit.script. Breaks torch.compile and is much less performant. Remove SpaceToDepthModule

2024-05-06 16:32:49 -07:00

loss

Fix #2132, remove use of _C.set_grad_enable. Line endings were messed up too

2024-04-09 09:00:23 -07:00

models

sbb vit weights on hub, testing

2024-05-10 17:15:01 -07:00

optim

…

scheduler

more

2024-04-02 10:25:30 +08:00

utils

Add AttentionExtract helper module

2024-05-04 14:10:00 -07:00

__init__.py

…

version.py

Prepping weight push, benchmarking.

2024-05-10 14:14:06 -07:00