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

Update sgdw for older pytorch

2023-12-11 12:10:29 -08:00

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

Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models

2022-12-06 15:00:06 -08:00

version.py

Prepping weight push, benchmarking.

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