Ross Wightman 66a0eb4673 Experimenting with tiny test models, how small can they go and be useful for regression tests?

2024-06-07 16:09:25 -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

Add ConvStem and MobileCLIP hybrid model for B variant. Add full norm disable support to ConvNormAct layers

2024-06-06 09:15:27 -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

Experimenting with tiny test models, how small can they go and be useful for regression tests?

2024-06-07 16:09:25 -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 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 version.py

2024-06-07 13:13:23 -07:00