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
...
Not six min
5 years ago
Ross Wightman
305a2db705
Update test_inference.py
...
Make the timeout 5-min for now, see if we can get a pass...
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
...
Setup Github Action to instantiate and run a forward pass with each registered model.
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.
...
Make small code refactoring
5 years ago
Ross Wightman
6cc11a8821
Merge pull request #141 from Animatory/fix_HRNet
...
Fixed HRNet modules
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
...
added 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
...
added efficientnet pruned weights
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
...
fix typo in eca
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.
...
* allow use of global pool arg, test-time-pooling
* clean checkpoints to just contain state dict, add 448 res checkpoints
* support DataParallel via lazy filter creation for JIT Downsample
* some minor formatting (mostly alignment) preferences
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
...
TResNet models
5 years ago
Ross Wightman
e15f979457
Merge pull request #123 from aclex/mobilenetv3_fix_feature_extraction
...
Merge changes in feature extraction interface to MobileNetV3
5 years ago
Alexey Chernov
bdb165a8a4
Merge changes in feature extraction interface to MobileNetV3
...
Experimental feature extraction interface seems to be changed
a little bit with the most up to date version apparently found
in EfficientNet class. Here these changes are added to
MobileNetV3 class to make it support it and work again, too.
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
mrT23
98a7403ed4
Merge pull request #1 from rwightman/master
...
merge master
5 years ago