Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
279 Commits
41 Branches
41 Tags
28 MiB
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '5b7cc16ac9'
${ noResults }
Commit Graph

279 Commits (5b7cc16ac915f9c162406ac855b29e6819296e64)
 

Author SHA1 Message Date
Ross Wightman 5b7cc16ac9 Add warning about using sync-bn with zero initialized BN layers. Fixes #54
5 years ago
Ross Wightman b18c19901e
Update README.md
5 years ago
Ross Wightman 12dbc74742 New ResNet50 JSD + RandAugment weights
5 years ago
Ross Wightman e39aae56b4
Update README.md
5 years ago
Ross Wightman 7a17ee9c3b Update README.md
5 years ago
Ross Wightman e42503e1d0 Update sotabench
5 years ago
Ross Wightman 2f41905ba5 Update ResNet50 weights to AuxMix trained 78.994 top-1. A few commentes re 'tiered_narrow' tn variant.
5 years ago
Ross Wightman cc0b1f4130 Add support to clean_checkpoint.py to remove aux_bn weights/biases from SplitBatchNorm
5 years ago
Ross Wightman 2a88412413
Update README.md
5 years ago
Ross Wightman d9a6a9d0af
Merge pull request #74 from rwightman/augmix-jsd
5 years ago
Ross Wightman 3eb4a96eda Update AugMix, JSD, etc comments and references
5 years ago
Ross Wightman a28117ea46 Add tiered narrow ResNet (tn) and weights for seresnext26tn_32x4d
5 years ago
Ross Wightman cfa951bceb
Update README.md
5 years ago
Ross Wightman 833066b540 A few minor things in SplitBN
5 years ago
Ross Wightman 7547119891 Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together
5 years ago
Ross Wightman 2e955cfd0c Update RandomErasing with some improved arg names, tweak to aspect range
5 years ago
Ross Wightman 3cc0f91e23 Fix augmix variable name scope overlap, default non-blended mode
5 years ago
Ross Wightman 7622015258
Update README.md
5 years ago
Ross Wightman ec0dd4053a Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko
5 years ago
Ross Wightman 8662454bd7 Plural for averaging script.
5 years ago
Ross Wightman 40fea63ebe Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
5 years ago
Ross Wightman 4666cc9aed Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script
5 years ago
Ross Wightman 53001dd292 ResNet / Res2Net additions:
5 years ago
Ross Wightman f96b3e5e92 InceptionResNetV2 torchscript compatible
5 years ago
Ross Wightman 19d93fe454 Add selecsls60 weights
5 years ago
Ross Wightman a7fe891a13
Update README.md
5 years ago
Ross Wightman e728d70831
Merge pull request #66 from rwightman/selecsls_updates
5 years ago
Ross Wightman 84ca3d1f4d Add SelecSLS to sotabench list
5 years ago
Ross Wightman 0062c15fb0 Update checkpoint url with modelzoo compatible ones.
5 years ago
Ross Wightman b5315e66b5 Streamline SelecSLS model without breaking checkpoint compat. Move cfg handling out of model class. Update feature/pooling behaviour to match current.
5 years ago
Ross Wightman d59a756c16 Run PyCharm autoformat on selecsls and change mix cap variables and model names to all lower
5 years ago
Ross Wightman fb3a0f4bb8
Merge pull request #65 from mehtadushy/selecsls
5 years ago
Ross Wightman 0554b7942c
Update README.md
5 years ago
Ross Wightman a4497afa21
Update sotabench.py
5 years ago
Ross Wightman 53f578e16f Update README with B3 training details
5 years ago
Ross Wightman 19fc205a4d Update comments on the new SE-ResNeXt26 models
5 years ago
Ross Wightman 8894a6bc08 Update sotabench.py with latest seresnext26 and efficientnet_b3 models
5 years ago
Ross Wightman acc3ed2b8c Add EfficientNet-B3 weights, trained from scratch with RA.
5 years ago
Dushyant Mehta 2404361f62 correct asset paths
5 years ago
Dushyant Mehta 31939311f6 Added SelecSLS models
5 years ago
rwightman 1f4498f217 Add ResNet deep tiered stem and model weights for seresnext26t_32x4d and seresnext26d_32x4d
5 years ago
Dushyant Mehta 32012a44fd Added SelecSLS model
5 years ago
Ross Wightman 73b78459dc Add update RandAugment MixNet-XL weights
5 years ago
Ross Wightman 3afc2a4dc0 Some cleanup/improvements to AugMix impl:
5 years ago
Ross Wightman 232ab7fb12 Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl
5 years ago
Ross Wightman ff8688ca3d
Merge pull request #62 from rwightman/reduce-bn
5 years ago
Ross Wightman 5719b493ad Missed update dist-bn logic for EMA model
5 years ago
Ross Wightman a435ea1327 Change reduce_bn to distribute_bn, add ability to choose between broadcast and reduce (mean). Add crop_pct arg to allow selecting validation crop while training.
5 years ago
Ross Wightman 3bff2b21dc Add support for keeping running bn stats the same across distributed training nodes before eval/save
5 years ago
Ross Wightman 0161de0127 Switch RandoErasing back to on GPU normal sampling
5 years ago