Commit Graph

8 Commits (1ea0be1b17d2238ae9c884b28ac38207f06f2f26)

Author SHA1 Message Date
Ross Wightman 1ea0be1b17 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)
1 year ago
Fredo Guan c43340ddd4
Davit std (#5)
1 year ago
Fredo Guan edea013dd1
Davit std (#3)
1 year ago
Fredo Guan 3bd96609c8
Davit (#1)
1 year ago