Ross Wightman 1b66ec7cf3 Fixup ViTamin, add hub weight reference

2024-06-03 17:14:03 -07:00

..

data

Add normalize flag to transforms factory, allow return of non-normalized native dtype torch.Tensors

2024-05-13 15:23:25 -07:00

layers

Credit earlier work with the same idea.

2024-05-16 22:50:34 +02:00

loss

…

models

Fixup ViTamin, add hub weight reference

2024-06-03 17:14:03 -07:00

optim

…

scheduler

…

utils

Add regex matching support to AttentionExtract. Add return_dict support to graph extractors and use returned output in AttentionExtractor

2024-05-22 14:33:39 -07:00

__init__.py

…

version.py

Version 1.0.4.dev0

2024-05-20 15:52:27 -07:00