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

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

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

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