mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
728 Commits
137 Branches
65 Tags
47 MiB
Commit Graph

728 Commits (5c7d2982346bd7faca4d083bd52dc9228dc9cdf7)
 

Author SHA1 Message Date
Ross Wightman 2f884a0ce5 Add resnest14, resnest26, and two of the abalation grouped resnest50 models 2020-05-12 12:21:52 -07:00
Ross Wightman f4cdc2ac31 Add ResNeSt models 2020-05-11 23:27:09 -07:00
Ross Wightman 780860d140 Add norm_act factory method, move JIT of norm layers to factory 2020-05-09 22:09:21 -07:00
Ross Wightman 14edacdf9a DenseNet converted to support ABN (norm + act) modules. Experimenting with EvoNorm, IABN 2020-05-09 18:26:41 -07:00
Ross Wightman e545bb9401
Update test_inference.py 
Not six min
 2020-05-07 10:15:49 -07:00
Ross Wightman 305a2db705
Update test_inference.py 
Make the timeout 5-min for now, see if we can get a pass...
 2020-05-07 10:14:24 -07:00
Ross Wightman 022ed001f3 Update DenseNet to latest in Torchvision (torchscript compat, checkpointing, proper init). Start adding ehanced configurability, stem options... 2020-05-07 09:57:48 -07:00
Ross Wightman 14e01b878c
Merge pull request #143 from michalwols/master 
Setup Github Action to instantiate and run a forward pass with each registered model.
 2020-05-07 09:47:13 -07:00
michal 8c77f14cae Install cpu version of torch on ubuntu 2020-05-07 01:09:16 -04:00
michal 69d725c9fe Basic forward pass test for all registered models 2020-05-07 00:20:58 -04:00
michal 8da43e0617 Install extra dependencies required by some models and log test durations 2020-05-07 00:20:34 -04:00
Mike 16f1f77b41
Add a test workflow for github actions 2020-05-06 23:21:50 -04:00
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
 2020-05-06 09:54:03 +03:00
Ross Wightman 6cc11a8821
Merge pull request #141 from Animatory/fix_HRNet 
Fixed HRNet modules
 2020-05-05 18:21:34 -07:00
Vyacheslav Shults f0eb021620 Replace all None by nn.Identity() in HRNet modules 2020-05-05 21:09:35 +03:00
Ross Wightman 3b72ebff51
Merge pull request #140 from yoniaflalo/PR_MultiEpochsDataLoader 
added MultiEpochsDataLoader
 2020-05-05 08:51:33 -07:00
AFLALO, Jonathan Isaac a7f570c9b7 added MultiEpochsDataLoader 2020-05-05 14:47:02 +03:00
Ross Wightman 8d8677e03b Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too. 2020-05-04 12:14:43 -07:00
Ross Wightman 353a79aeba
Update README.md 2020-05-03 21:24:49 -07:00
Ross Wightman c9b6f415ca Bump version for Pypi release 2020-05-03 21:18:47 -07:00
Ross Wightman 31f4c126ec Add EfficientNet pruned models to results files 2020-05-03 20:15:11 -07:00
Ross Wightman 375f3e5321 Fix pruned txt files not being installed during pip install 2020-05-03 19:06:14 -07:00
Ross Wightman 8ec554b82e
Merge pull request #136 from yoniaflalo/adding_effnet_pruned 
added efficientnet pruned weights
 2020-05-03 07:14:14 -07:00
AFLALO, Jonathan Isaac 9c15d57505 added efficientnet pruned weights 2020-05-03 09:29:45 +03:00
Ross Wightman a4d20a1cb8
Update README.md 2020-05-01 19:22:35 -07:00
Ross Wightman ea300709f0 Fix model create fn not passing num_classes through. Fix #135 2020-05-01 18:37:03 -07:00
Ross Wightman 779cb0fcc0 Update results with new models 2020-05-01 17:29:29 -07:00
Ross Wightman 2c438c4e0d Merge branch 'master' of github.com:rwightman/pytorch-models 2020-05-01 17:03:17 -07:00
Ross Wightman 28739bb721 Merge branch 'VRandme-blur' 2020-05-01 17:01:43 -07:00
Ross Wightman 2681a8d618 Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop 2020-05-01 17:00:21 -07:00
Ross Wightman 9590f301a9 Merge branch 'blur' of https://github.com/VRandme/pytorch-image-models into VRandme-blur 2020-05-01 15:19:57 -07:00
Ross Wightman a5220ad3aa
Merge pull request #125 from Separius/patch-1 
fix typo in eca
 2020-05-01 15:11:56 -07:00
Ross Wightman 7a9942a75e Merge branch 'yoniaflalo-adding_ECA_resnet' 2020-05-01 15:10:23 -07:00
Ross Wightman 0834fbc01c Move pruned model adapt strings to separate txt files. A few minor formatting alignment tweaks. 2020-05-01 15:09:20 -07:00
Ross Wightman ba793f5c1c Merge branch 'adding_ECA_resnet' of https://github.com/yoniaflalo/pytorch-image-models into yoniaflalo-adding_ECA_resnet 2020-04-29 16:43:02 -07:00
AFLALO, Jonathan Isaac 07f19dd699 added eca resnet 2020-04-29 19:09:55 +03:00
Ross Wightman 1d4ac1b165 Bump version for pypi release. Fix #130 2020-04-27 18:02:40 -07:00
Ross Wightman be7c784d21 Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference 2020-04-27 17:50:19 -07:00
Ross Wightman e3a98171b2 Remove some CRLF that made it into TResNet merge 2020-04-27 13:23:41 -07:00
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
 2020-04-27 13:21:05 -07:00
Ross Wightman 64fe37d008 Remove old mean/std helper, rely fully on cmd line or default_cfg now. Fixes #126 2020-04-27 12:32:55 -07:00
Ross Wightman 02a30411ad Replace fp16 with amp support for validate.py script 2020-04-27 12:31:36 -07:00
Ross Wightman e6f24e5578 Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults 2020-04-25 19:42:13 -07:00
Sepehr Sameni 20290b5b56
fix typo in eca 2020-04-14 15:20:35 +04:30
Ross Wightman ebf82b84ac
Merge pull request #122 from mrT23/master 
TResNet models
 2020-04-12 18:23:46 -07:00
Ross Wightman e15f979457
Merge pull request #123 from aclex/mobilenetv3_fix_feature_extraction 
Merge changes in feature extraction interface to MobileNetV3
 2020-04-12 18:17:13 -07:00
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.
 2020-04-13 02:02:14 +03:00
talrid 1c92755bb6 Merge branch 'master' of https://github.com/mrT23/pytorch-image-models 2020-04-12 19:09:39 +03:00
talrid 011a11d987 TResNet models 2020-04-12 19:09:14 +03:00
talrid 8a63c1add8 finalizing 2020-04-12 19:08:35 +03:00