Commit Graph

7 Commits (7a13be67a54e02aa3f16f9ca57cadd6a501f3373)

Author SHA1 Message Date
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