You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm
Ross Wightman ea9c9550b2
Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only.
3 years ago
..
data Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update 3 years ago
loss asymmetric_loss 4 years ago
models Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only. 3 years ago
optim Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface. 3 years ago
scheduler Merge branch 'logger' into features. Change 'logger' to '_logger'. 4 years ago
utils Add GIST to docstring for quick access 3 years ago
__init__.py Monster commit, activation refactor, VoVNet, norm_act improvements, more 4 years ago
version.py Add ECA-NFNet-L0 weights and update model name. Update README and bump version to 0.4.6 3 years ago