Commit Graph

8 Commits (b995ca3c314598b2b91ca1755232652c6df07335)

Author SHA1 Message Date
Ross Wightman 7d9e321b76 Improve tracing of window attn models with simpler reshape logic
1 year ago
Ross Wightman 6f28b562c6 Factor NormMlpClassifierHead from MaxxViT and use across MaxxViT / ConvNeXt / DaViT, refactor some type hints & comments
1 year ago
Ross Wightman 9a53c3f727 Finalize DaViT, some formatting and modelling simplifications (separate PatchEmbed to Stem + Downsample, weights on HF hub.
1 year ago
Fredo Guan 81ca323751
Davit update formatting and fix grad checkpointing (#7)
1 year ago
Fredo Guan 10b3f696b4
Davit std (#6)
2 years ago
Fredo Guan c43340ddd4
Davit std (#5)
2 years ago
Fredo Guan edea013dd1
Davit std (#3)
2 years ago
Fredo Guan 3bd96609c8
Davit (#1)
2 years ago