Commit Graph

177 Commits (ff94ffce613fc5cbe263ef585aafe5d4aee95a7c)

Author SHA1 Message Date
Ross Wightman 0ea53cecc3 Merge branch 'master' into densenet_update_and_more
5 years ago
Ross Wightman 6441e9cc1b Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.
5 years ago
Ross Wightman afb6bd0669 Add backward and default_cfg tests and fix a few issues found. Fix #153
5 years ago
Ross Wightman 50658b9a67 Add RegNet models and weights
5 years ago
Ross Wightman 7df83258c9 Merge branch 'master' into densenet_update_and_more
5 years ago
Ross Wightman 1904ed8fec Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups
5 years ago
Ross Wightman 17270c69b9 Remove annoying InceptionV3 dependency on scipy and insanely slow trunc_norm init. Bring InceptionV3 code into this codebase and use upcoming torch trunch_norm_ init.
5 years ago
Ross Wightman c4ca016656
Merge pull request #145 from rwightman/resnest
5 years ago
Ross Wightman 208e7912f7 Missed one of the abalation model entrypoints, update README
5 years ago
Ross Wightman 9cc289f18c Exclude EfficientNet-L2 models from test
5 years ago
Ross Wightman 2f884a0ce5 Add resnest14, resnest26, and two of the abalation grouped resnest50 models
5 years ago
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 022ed001f3 Update DenseNet to latest in Torchvision (torchscript compat, checkpointing, proper init). Start adding ehanced configurability, stem options...
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
Vyacheslav Shults f0eb021620 Replace all None by nn.Identity() in HRNet modules
5 years ago
Ross Wightman 8d8677e03b Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.
5 years ago
AFLALO, Jonathan Isaac 9c15d57505 added efficientnet pruned weights
5 years ago
Ross Wightman ea300709f0 Fix model create fn not passing num_classes through. Fix #135
5 years ago
Ross Wightman 2c438c4e0d Merge branch 'master' of github.com:rwightman/pytorch-models
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 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 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
Sepehr Sameni 20290b5b56
fix typo in eca
5 years ago
Ross Wightman ebf82b84ac
Merge pull request #122 from mrT23/master
5 years ago
Alexey Chernov bdb165a8a4 Merge changes in feature extraction interface to MobileNetV3
5 years ago
talrid 8a63c1add8 finalizing
5 years ago
talrid 6209146738 TResNet models
5 years ago
Ross Wightman 1a8f5900ab Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
5 years ago
Chris Ha 06a50a94a8 Fix minor typos in create_attn.py and resnet.py
5 years ago
Ross Wightman 1a9ab07307 One too many changes at a time, fix missing C
5 years ago
Ross Wightman f17b42bc33 Blur filter no longer a buffer
5 years ago
Ross Wightman 6cdeca24a3 Some cleanup and fixes for initial BlurPool impl. Still some testing and tweaks to go...
5 years ago
Chris Ha acd1b6cccd Implement Functional Blur on resnet.py
5 years ago
Chris Ha ce3d82b58b Update blurpool.py
5 years ago
Chris Ha 3a287a6e76 Create blurpool.py
5 years ago
Ross Wightman c99a5abed4
Merge pull request #115 from rwightman/mobilenetv2-experiment
5 years ago
Ross Wightman e34074b4da Add final weights for MobileNet-V2 experiments
5 years ago
Ross Wightman a6a5565de7 Fiddling...
5 years ago
Ross Wightman 5a16c533ff Add better resnext50_32x4d weights trained by andravin
5 years ago
Ross Wightman bc998cad91 Experimenting with some MobileNetV2 variations to compare against EfficientNet-Lite
5 years ago
Ross Wightman 3406e582cf Add EfficientNet-Lite results, update README
5 years ago
Ross Wightman bd05258f7b EfficientNet-Lite model added w/ converted checkpoints, validation in progress...
5 years ago