Ross Wightman 28d76a97db Mixed up kernel size for last blocks in mnv4-conv-small

2024-05-24 11:50:42 -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

torchscript typing fixes

2024-05-23 11:43:05 -07:00

loss

…

models

Mixed up kernel size for last blocks in mnv4-conv-small

2024-05-24 11:50:42 -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