Commit Graph

654 Commits (fd9061dbf74df86431128ce0ca0547858c7600ca)
 

Author SHA1 Message Date
Ross Wightman 7be299504f Add missing feature_info() on MobileNetV3, make hook feature output order/type consistent with bottleneck (list, decreasing fmap size)
5 years ago
Ross Wightman 88129b2569 Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.
5 years ago
Ross Wightman f28170df3f Fix an an untested change, remove a debug print
5 years ago
Ross Wightman eb7653614f Monster commit, activation refactor, VoVNet, norm_act improvements, more
5 years ago
Ross Wightman ff94ffce61 Merge branch 'master' into densenet_update_and_more
5 years ago
Ross Wightman 5966654052 Update requirements so PyTorch 1.4 is min, add separate sotabench reqs to force 1.4 and prevent error with 1.5
5 years ago
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 d79ac48626 Update sotabench.py
5 years ago
Ross Wightman e881383b4e
Merge pull request #154 from rwightman/tests_bugfixes
5 years ago
Ross Wightman 4d13db538f Add x48d ResNext101s to test exclude for ubuntu
5 years ago
Ross Wightman 4212cd3b9f Another attempt at getting Ubuntu test runner to work
5 years ago
Ross Wightman 20329f2630 Bring down test resolutions to see if we can at least do a fwd on the L2 models
5 years ago
Ross Wightman 3873ea710e Minor test change
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 ea2e59ca36
Merge pull request #150 from rwightman/regnet
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 dab9935b36
Merge pull request #148 from rwightman/drop_block_improve
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 63addb741f
Merge pull request #146 from rwightman/inceptionv3_fix
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 5bd1ad13e7 Refactor test indent
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 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