Commit Graph

8 Commits (a5b01ec04e7ba78d0b5ab5c3f2f43a356562a130)

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