Commit Graph

1277 Commits (1420c118dfa1ba151a9cbd76f08db2701da23bbe)
 

Author SHA1 Message Date
Ross Wightman f4cdc2ac31 Add ResNeSt models
5 years ago
Ross Wightman 780860d140 Add norm_act factory method, move JIT of norm layers to factory
5 years ago
Ross Wightman 14edacdf9a DenseNet converted to support ABN (norm + act) modules. Experimenting with EvoNorm, IABN
5 years ago
Ross Wightman e545bb9401
Update test_inference.py
5 years ago
Ross Wightman 305a2db705
Update test_inference.py
5 years ago
Ross Wightman 022ed001f3 Update DenseNet to latest in Torchvision (torchscript compat, checkpointing, proper init). Start adding ehanced configurability, stem options...
5 years ago
Ross Wightman 14e01b878c
Merge pull request #143 from michalwols/master
5 years ago
michal 8c77f14cae Install cpu version of torch on ubuntu
5 years ago
michal 69d725c9fe Basic forward pass test for all registered models
5 years ago
michal 8da43e0617 Install extra dependencies required by some models and log test durations
5 years ago
Mike 16f1f77b41
Add a test workflow for github actions
5 years ago
Vyacheslav Shults a7ebe09029 Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.
5 years ago
Ross Wightman 6cc11a8821
Merge pull request #141 from Animatory/fix_HRNet
5 years ago
Vyacheslav Shults f0eb021620 Replace all None by nn.Identity() in HRNet modules
5 years ago
Ross Wightman 3b72ebff51
Merge pull request #140 from yoniaflalo/PR_MultiEpochsDataLoader
5 years ago
AFLALO, Jonathan Isaac a7f570c9b7 added MultiEpochsDataLoader
5 years ago
Ross Wightman 8d8677e03b Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.
5 years ago
Ross Wightman 353a79aeba
Update README.md
5 years ago
Ross Wightman c9b6f415ca Bump version for Pypi release
5 years ago
Ross Wightman 31f4c126ec Add EfficientNet pruned models to results files
5 years ago
Ross Wightman 375f3e5321 Fix pruned txt files not being installed during pip install
5 years ago
Ross Wightman 8ec554b82e
Merge pull request #136 from yoniaflalo/adding_effnet_pruned
5 years ago
AFLALO, Jonathan Isaac 9c15d57505 added efficientnet pruned weights
5 years ago
Ross Wightman a4d20a1cb8
Update README.md
5 years ago
Ross Wightman ea300709f0 Fix model create fn not passing num_classes through. Fix #135
5 years ago
Ross Wightman 779cb0fcc0 Update results with new models
5 years ago
Ross Wightman 2c438c4e0d Merge branch 'master' of github.com:rwightman/pytorch-models
5 years ago
Ross Wightman 28739bb721 Merge branch 'VRandme-blur'
5 years ago
Ross Wightman 2681a8d618 Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop
5 years ago
Ross Wightman 9590f301a9 Merge branch 'blur' of https://github.com/VRandme/pytorch-image-models into VRandme-blur
5 years ago
Ross Wightman a5220ad3aa
Merge pull request #125 from Separius/patch-1
5 years ago
Ross Wightman 7a9942a75e Merge branch 'yoniaflalo-adding_ECA_resnet'
5 years ago
Ross Wightman 0834fbc01c Move pruned model adapt strings to separate txt files. A few minor formatting alignment tweaks.
5 years ago
Ross Wightman ba793f5c1c Merge branch 'adding_ECA_resnet' of https://github.com/yoniaflalo/pytorch-image-models into yoniaflalo-adding_ECA_resnet
5 years ago
AFLALO, Jonathan Isaac 07f19dd699 added eca resnet
5 years ago
Ross Wightman 1d4ac1b165 Bump version for pypi release. Fix #130
5 years ago
Ross Wightman be7c784d21 Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference
5 years ago
Ross Wightman e3a98171b2 Remove some CRLF that made it into TResNet merge
5 years ago
Ross Wightman 0004f37d25 Some TResNet cleanup.
5 years ago
Ross Wightman 64fe37d008 Remove old mean/std helper, rely fully on cmd line or default_cfg now. Fixes #126
5 years ago
Ross Wightman 02a30411ad Replace fp16 with amp support for validate.py script
5 years ago
Ross Wightman e6f24e5578 Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults
5 years ago
Sepehr Sameni 20290b5b56
fix typo in eca
5 years ago
Ross Wightman ebf82b84ac
Merge pull request #122 from mrT23/master
5 years ago
Ross Wightman e15f979457
Merge pull request #123 from aclex/mobilenetv3_fix_feature_extraction
5 years ago
Alexey Chernov bdb165a8a4 Merge changes in feature extraction interface to MobileNetV3
5 years ago
talrid 1c92755bb6 Merge branch 'master' of https://github.com/mrT23/pytorch-image-models
5 years ago
talrid 011a11d987 TResNet models
5 years ago
talrid 8a63c1add8 finalizing
5 years ago
talrid 6209146738 TResNet models
5 years ago