Commit Graph

1701 Commits (316bdf895590a66825b319e3901f837709f7ded0)
 

Author SHA1 Message Date
Ross Wightman 883fa2eeaa Add fine-tuned B/16 224x224 in1k clip models
2 years ago
Ross Wightman 9a3d2ac2d5 Add latest CLIP ViT fine-tune pretrained configs / model entrypt updates
2 years ago
Ross Wightman 42bbbddee9 Add missing model config
2 years ago
Ross Wightman def68befa7 Updating vit model defs for mult-weight support trial (vit first). Prepping for CLIP (laion2b and openai) fine-tuned weights.
2 years ago
Ross Wightman 0dadb4a6e9 Initial multi-weight support, handled so old pretraing config handling co-exists with new tags.
2 years ago
Ross Wightman e8ddc6865c
Update torch to 1.13 and python to 3.10 in test workflow (#1562)
2 years ago
Ross Wightman 98c5e78226
Merge pull request #1564 from HongxinXiang/main
2 years ago
hongxin xiang 653bdc7105 Fix comment: https://github.com/rwightman/pytorch-image-models/pull/1564#issuecomment-1326743424
2 years ago
hongxin xiang bdc9fad638 Fix compatible BUG: QMNIST and ImageNet datasets do not exist in torchvision 0.10.1.
2 years ago
Ross Wightman 25ffac6880
Merge pull request #1547 from Wauplin/1546-fix-hf-hub-integration
2 years ago
Wauplin 9b114754db refactor push_to_hub helper
2 years ago
Wauplin ae0a0db7de Create repo before cloning with Repository.clone_from
2 years ago
Ross Wightman 8ff45e41f7 Fix xdist dependency change
2 years ago
Ross Wightman 4714a4910e
Merge pull request #1525 from TianyiFranklinWang/main
2 years ago
Ross Wightman f856f8a2e0
Merge pull request #1527 from klae01/patch-1
2 years ago
klae01 ddd6361904
Update train.py
2 years ago
NPU-Franklin 9152b10478
✏️ fix typo
2 years ago
Ross Wightman 4201c718e9
Merge pull request #1522 from hova88/main
2 years ago
hova88 29baf32327 fix typo : miss back quote
2 years ago
Ross Wightman 803254bb40 Fix spacing misalignment for fast norm path in LayerNorm modules
2 years ago
Ross Wightman 475ecdfa3d cast env var args for dataset readers to int
2 years ago
Ross Wightman 44cea489fb
Merge pull request #1502 from infomon/main
2 years ago
Simon Schrodi aceb79e002 Fix typo
2 years ago
Ross Wightman 8a1ba67a50
Merge pull request #1500 from nguyenhoan1988/main
2 years ago
Hoan Nguyen 39190f5f44
Remove inplace operators when calculating the loss
2 years ago
Ross Wightman 91fc4c7ed1
Update tests.yml
2 years ago
Ross Wightman 26c400bf1e
Merge pull request #1498 from nateraw/docs-deploy-from-main
2 years ago
nateraw cf549c28b1 🎨 deploy docs from main not master
2 years ago
Ross Wightman 3e57377f7a
Update README.md
2 years ago
Ross Wightman 6635bc3f7d
Merge pull request #1479 from rwightman/script_cleanup
2 years ago
Ross Wightman 0e6023f032
Merge pull request #1381 from ChristophReich1996/master
2 years ago
Ross Wightman 66f4af7090 Merge remote-tracking branch 'origin/master' into script_cleanup
2 years ago
Ross Wightman eb333d6641 Update validate.py to use updated amp args for impl/dtype
2 years ago
Ross Wightman d3961536c9 comment some debug logs for WDS dataset
2 years ago
Ross Wightman e9dccc918c Rename dataset/parsers -> dataset/readers, create_parser to create_reader, etc
2 years ago
Ross Wightman 8c28363dc9 Version 0.7.dev0 for master
2 years ago
Ross Wightman 1ae6ba62cd
Merge pull request #1494 from nateraw/add-dev-suffix-to-tag
2 years ago
nateraw 30bafd7347 🔖 add dev suffix to version tag
2 years ago
Ross Wightman 94aed25224
Merge pull request #1493 from nateraw/fix-hf-docs-build
2 years ago
nateraw edceac4aaf 💚 fix ci build
2 years ago
Ross Wightman 2f041b84da
Merge pull request #1473 from nateraw/hfdocs
2 years ago
Ross Wightman f67a7ee8bd Set num_workers in Iterable WDS/TFDS datasets early so sample estimate is correct
2 years ago
Ross Wightman bb49d02aae
Update README.md
2 years ago
Ross Wightman b944546200
Update README.md
2 years ago
Ross Wightman cea8df3d0c Version 0.6.12
2 years ago
Ross Wightman 9914f744dc Add more maxxvit weights includ ConvNeXt conv block based experiments.
2 years ago
Ross Wightman 285771972e Change --amp flags, no more --apex-amp and --native-amp, add --amp-impl to select apex, and --amp-dtype to allow bfloat16 AMP dtype
2 years ago
Ross Wightman b1b024dfed Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168
2 years ago
Ross Wightman 4f18d6dc5f Fix logs in WDS parser
2 years ago
nateraw af0c644c5d 👷 remove --not_python_module flag
2 years ago