Ross Wightman a0bb5b4a44 Missing stem_kernel_size argument in EfficientNetFeatures		2024-06-14 13:39:31 -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	Fix rotary embed version of attn pool. Bit of cleanup/naming	2024-06-11 23:49:17 -07:00
loss	…
models	Missing stem_kernel_size argument in EfficientNetFeatures	2024-06-14 13:39:31 -07:00
optim	…
scheduler	more	2024-04-02 10:25:30 +08:00
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	Update README.md, bump dev version 1.0.6	2024-06-12 12:35:44 -07:00