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

92 Commits (9f11b4e8a25495874d84a56d4ca11af191a01324)

Author SHA1 Message Date
Ross Wightman 9f11b4e8a2 Add ConvBnAct layer to parallel integrated SelectKernelConv, add support for DropPath and DropBlock to ResNet base and SK blocks
6 years ago
Ross Wightman cefc9b7761 Move SelectKernelConv to conv2d_layers and more
6 years ago
Ross Wightman 9abe610931 Used wrong channel var for split
6 years ago
Ross Wightman 58e28dc7e7 Move Selective Kernel blocks/convs to their own sknet.py file
6 years ago
Ross Wightman a93bae6dc5 A SelectiveKernelBasicBlock for more experiments
6 years ago
Ross Wightman ad087b4b17 Missed bias=False in selection conv
6 years ago
Ross Wightman c8b3d6b81a Initial impl of Selective Kernel Networks. Very much a WIP.
6 years ago
Ross Wightman 91534522f9 Add newly added TF ported EfficientNet-B8 weights (RandAugment)
6 years ago
Ross Wightman 12dbc74742 New ResNet50 JSD + RandAugment weights
6 years ago
Ross Wightman 2f41905ba5 Update ResNet50 weights to AuxMix trained 78.994 top-1. A few commentes re 'tiered_narrow' tn variant.
6 years ago
Ross Wightman d9a6a9d0af
Merge pull request #74 from rwightman/augmix-jsd
6 years ago
Ross Wightman 3eb4a96eda Update AugMix, JSD, etc comments and references
6 years ago
Ross Wightman a28117ea46 Add tiered narrow ResNet (tn) and weights for seresnext26tn_32x4d
6 years ago
Ross Wightman 833066b540 A few minor things in SplitBN
6 years ago
Ross Wightman 7547119891 Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together
6 years ago
Ross Wightman ec0dd4053a Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko
6 years ago
Ross Wightman 40fea63ebe Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
6 years ago
Ross Wightman 53001dd292 ResNet / Res2Net additions:
6 years ago
Ross Wightman f96b3e5e92 InceptionResNetV2 torchscript compatible
6 years ago
Ross Wightman 19d93fe454 Add selecsls60 weights
6 years ago
Ross Wightman 0062c15fb0 Update checkpoint url with modelzoo compatible ones.
6 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.
6 years ago
Ross Wightman d59a756c16 Run PyCharm autoformat on selecsls and change mix cap variables and model names to all lower
6 years ago
Ross Wightman fb3a0f4bb8
Merge pull request #65 from mehtadushy/selecsls
6 years ago
Ross Wightman 19fc205a4d Update comments on the new SE-ResNeXt26 models
6 years ago
Ross Wightman acc3ed2b8c Add EfficientNet-B3 weights, trained from scratch with RA.
6 years ago
Dushyant Mehta 2404361f62 correct asset paths
6 years ago
Dushyant Mehta 31939311f6 Added SelecSLS models
6 years ago
rwightman 1f4498f217 Add ResNet deep tiered stem and model weights for seresnext26t_32x4d and seresnext26d_32x4d
6 years ago
Dushyant Mehta 32012a44fd Added SelecSLS model
6 years ago
Ross Wightman 73b78459dc Add update RandAugment MixNet-XL weights
6 years ago
Ross Wightman ff421e5e09 New PyTorch trained EfficientNet-B2 weights with my RandAugment impl
6 years ago
Ross Wightman 3bef524f9c Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.
6 years ago
Ross Wightman 6ca0828166 Update EfficientNet comments, MobileNetV3 non-TF create fns, fix factory arg checks, bump PyTorch version req to 1.2
6 years ago
Ross Wightman eccbadca74 Update EfficientNet comments
6 years ago
Ross Wightman 902d32fb16 Renamed gen_efficientnet.py -> efficientnet.py
6 years ago
Ross Wightman 5a0a8de7e3 ResNet updates:
6 years ago
Ross Wightman a39cc43374 Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
6 years ago
Ross Wightman ad93347548 Initial HRNet classification model commit
6 years ago
Ross Wightman 2393708650 Missed stashing of out_indices in model
6 years ago
Ross Wightman 35e8f0c5e7 Fixup a few comments, add PyTorch version aware Flatten and finish as_sequential for GenEfficientNet
6 years ago
Ross Wightman 7ac6db4543 Missed activations.py
6 years ago
Ross Wightman 506df0e3d0 Add CondConv support for EfficientNet into WIP for GenEfficientNet Feature extraction setup
6 years ago
Ross Wightman 576d360f20 Bring in JIT version of optimized swish activation from gen_efficientnet as default (while working on feature extraction functionality here).
6 years ago
Ross Wightman 7b83e67f77 Pass drop connect arg through to EfficientNet models
6 years ago
Ross Wightman 0d58c50fb1 Add TF RandAug weights for B5/B7 EfficientNet models.
6 years ago
Ross Wightman c099374771 Map pretrained checkpoint to cpu to avoid issue with some pretrained checkpoints still having CUDA tensors. Fixes #42
6 years ago
Ross Wightman b93fcf0708 Add Facebook Research Semi-Supervised and Semi-Weakly Supervised ResNet model weights.
6 years ago
Ross Wightman a9eb484835 Add memory efficient Swish impl
6 years ago
rwightman d3ba34ee7e Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup.
6 years ago