Commit Graph

785 Commits (288682796f4a18760f1276dd2d184f297fdf5182)
 

Author SHA1 Message Date
Ross Wightman 288682796f Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7
4 years ago
Ross Wightman ea9c9550b2 Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only.
4 years ago
Ross Wightman a5310a3451 Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update
4 years ago
Ross Wightman 7953e5d11a Fix pos_embed scaling for ViT and num_classes != 1000 for pretrained distilled deit and pit models. Fix #426 and fix #433
4 years ago
Ross Wightman a760a4c3f4 Some ViT cleanup, merge distilled model with main, fixup torchscript support for distilled models
4 years ago
Ross Wightman 0dfc5a66bb Add PiT model from https://github.com/naver-ai/pit
4 years ago
Ross Wightman 1ad1645a50 Merge branch 'contrastive-master'
4 years ago
Ross Wightman 51febd869b Small tweak to tests for tnt model, reorder model imports.
4 years ago
Ross Wightman b27a4e0d88 Merge branch 'master' of https://github.com/contrastive/pytorch-image-models into contrastive-master
4 years ago
Ross Wightman 2319cbbff2
Merge pull request #525 from amaarora/spp
4 years ago
Aman Arora 6b18061773 Add GIST to docstring for quick access
4 years ago
contrastive 809271b0f3
Update test_models.py
4 years ago
contrastive de86314655 Update TNT
4 years ago
Aman Arora 92b1db9a79 update docstrings and add check on and
4 years ago
Aman Arora b85be24054 update to work with fnmatch
4 years ago
contrastive cfc15283a4 Update TNT url
4 years ago
contrastive 4a09bc851e Add TNT model
4 years ago
Aman Arora 20626e8387 Add to extract stats for SPP
4 years ago
Ross Wightman a2727c1bf7
Merge pull request #510 from rwightman/dependabot/pip/jinja2-2.11.3
4 years ago
Ross Wightman e2e3290fbf Add '--experiment' to train args for fixed exp name if desired, 'train' not added to output folder if specified.
4 years ago
Ross Wightman cf5fec5047 Cleanup experimental vit weight init a bit
4 years ago
dependabot[bot] b36099ace3
Bump jinja2 from 2.11.2 to 2.11.3
4 years ago
Ross Wightman f42f1df26c Improve evenness of per-worker split for validation set with TFDS
4 years ago
Ross Wightman cbcb76d72c Should have included Conv2d layers in original weight init. Lets see what the impact is...
4 years ago
Ross Wightman 4de57ccf01 Add weight init scheme that's closer to JAX impl
4 years ago
Ross Wightman 14ac4abf74 Change huggingface hub revision delimiter to '@', add hf_hub reference for eca_nfnet_l0 model as an example.
4 years ago
Ross Wightman 740f32c96a Add ECA-NFNet-L0 weights and update model name. Update README and bump version to 0.4.6
4 years ago
Ross Wightman 5e2e4e7fb6
Merge pull request #502 from khatami-mehrdad/efficientnet_dg
4 years ago
Ross Wightman 3eac7dc5a3
Merge pull request #501 from rwightman/hf_hub_revisit
4 years ago
Ross Wightman 45c048ba13 A few minor fixes and bit more cleanup on the huggingface hub integration.
4 years ago
Mehrdad 97f49d163c adding efficientnet_el, efficientnet_es_prunedefficientnet_es_pruned and efficientnet_el_pruned pretrained models to timm/models
4 years ago
Ross Wightman ead80d33c5 Fix typo, naming consistency
4 years ago
Ross Wightman d584e7f617 Support for huggingface hub via create_model and default_cfgs.
4 years ago
Ross Wightman e5ba5dcbc9 Merge branch 'yoniaflalo-adding_Hardcore_NAS'
4 years ago
Ross Wightman 1f799afcb3 Remove capital letters in model names
4 years ago
Ross Wightman 52b5d0ad0a Merge branch 'adding_Hardcore_NAS' of https://github.com/yoniaflalo/pytorch-image-models into yoniaflalo-adding_Hardcore_NAS
4 years ago
AFLALO, Jonathan Isaac 82c1fe958a added hardcorenas model
4 years ago
Ross Wightman 06aa926fbf Tweak docs model summaries / pages names and layout
4 years ago
Ross Wightman ac9a2c978d Update docs requirements
4 years ago
Ross Wightman dda8fd80d2 Merge branch 'gcucurull-master'
4 years ago
Ross Wightman 74fcd4c50e Update finetune detail, rerun model doc gen.
4 years ago
Ross Wightman e6f5617dcc Fix a few typos and consistency issues in model docs
4 years ago
Ross Wightman 5392c0a4b3 Rerun modelindex generate script
4 years ago
Ross Wightman 36a5648208 Merge branch 'master' of https://github.com/gcucurull/pytorch-image-models into gcucurull-master
4 years ago
Ross Wightman dfae015c5d Move modelindex files into doc tree, add headings to model templates. Tweak a few details.
4 years ago
Marcin Kardas 8936356139 Add results to model-index
4 years ago
Guillem Cucurull 4d26eed4d4 fix typo in urls
4 years ago
Guillem Cucurull 3f6501c1ad Add short comment
4 years ago
Guillem Cucurull 9e74125276 Model index files and script to generate it
4 years ago
Ross Wightman 5c7d298234 Update README.md, documentation, add TOC, sponsors, and links to timmdocs and paperswithcode
4 years ago