Ross Wightman 086bd55a94 Add EfficientFormer-V2, refactor EfficientFormer and Levit for more uniformity across the 3 related arch. Add features_out support to levit conv models and efficientformer_v2. All weights on hub.

2023-02-03 14:12:29 -08:00

..

data

Cleanup resolve data config fns, add 'model' variant that takes model as first arg, make 'args' arg optional in original fn

2023-01-20 14:47:55 -08:00

layers

Add EfficientFormer-V2, refactor EfficientFormer and Levit for more uniformity across the 3 related arch. Add features_out support to levit conv models and efficientformer_v2. All weights on hub.

2023-02-03 14:12:29 -08:00

loss

…

models

Add EfficientFormer-V2, refactor EfficientFormer and Levit for more uniformity across the 3 related arch. Add features_out support to levit conv models and efficientformer_v2. All weights on hub.

2023-02-03 14:12:29 -08:00

optim

Add Adan to __init__.py

2022-12-15 11:37:29 +08:00

scheduler

…

utils

Davit update formatting and fix grad checkpointing (#7)

2023-01-15 14:34:56 -08:00

__init__.py

…

version.py

version 0.8.8dev0

2023-01-25 18:02:48 -08:00