mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
1,056 Commits 141 Branches 65 Tags
Commit Graph

1056 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Ross Wightman
a69c0e04f0 Fix pool size in cspnet 2020-07-27 13:44:02 -07:00
Ross Wightman
14ef7a0dd6 Rename csp.py -> cspnet.py 2020-07-27 11:15:07 -07:00
Ross Wightman
ec37008432 Add pretrained weight links to CSPNet for cspdarknet53, cspresnext50 2020-07-27 11:13:21 -07:00
Sangdoo Yun
e93e571f7a Add adamp and 'sgdp' optimizers. 
Update requirements.txt

Update optim_factory.py

Add `adamp` optimizer

Update __init__.py

copy files of adamp & sgdp

Create adamp.py

Update __init__.py

Create sgdp.py

Update optim_factory.py

Update optim_factory.py

Update requirements.txt

Update adamp.py

Update sgdp.py

Update sgdp.py

Update adamp.py
 2020-07-25 15:33:20 -07:00
Ross Wightman
0915beddad Add autosquash workflow 2020-07-25 14:26:30 -07:00
Ross Wightman
08016e839d Cleanup FeatureInfo getters, add TF models sourced Xception41/65/71 weights 2020-07-24 17:59:21 -07:00
Ross Wightman
7ba5a384d3 Add ReXNet w/ remapped weights, feature support 2020-07-23 10:28:57 -07:00
Ross Wightman
c9d54bc1c3 Add HRNet feature extraction, fix senet type, lower feature testing res to 96x96 2020-07-21 17:39:29 -07:00
Ross Wightman
2ac663f340 Add feature support to legacy senets, add 32x32 resnext models to exclude list for feature testing. 2020-07-21 11:15:30 -07:00
Ross Wightman
701dba303e Again 2020-07-21 08:53:51 -07:00
Ross Wightman
648ba41b4b Try again with the worfklow threads 2020-07-21 08:52:56 -07:00
Ross Wightman
52b6e72e86 Try defining num threads for github workflow tests explicity (2) 2020-07-21 08:49:11 -07:00
Ross Wightman
c146b54abc Cleanup EfficientNet/MobileNetV3 feature extraction a bit, only two tap locations now, small mobilenetv3 models work 2020-07-21 01:21:38 -07:00
Ross Wightman
68fd8a267b Merge branch 'master' into features 2020-07-20 16:11:38 -07:00
Ross Wightman
4e61c6a12d Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 2020-07-20 16:10:31 -07:00
Ross Wightman
6eec3fb4a4 Move FeatureHooks into features.py, switch EfficientNet, MobileNetV3 to use build model helper 2020-07-19 15:00:43 -07:00
Ross Wightman
9eba134d79 More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params. 2020-07-19 14:02:02 -07:00
Ross Wightman
298fba09ac Back out some activation hacks trialing upcoming pytorch changes 2020-07-17 18:41:37 -07:00
Ross Wightman
3b9004bef9 Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 2020-07-17 17:54:26 -07:00
Ross Wightman
e2cc481310 Update CSP ResNets for cross expansion without activation. Fix VovNet IABN compatibility with fixed activation arg. 2020-07-13 16:24:55 -07:00
Ross Wightman
3b6cce4c95 Add initial impl of CrossStagePartial networks, yet to be trained, not quite the same as darknet cfgs. 2020-07-13 15:01:06 -07:00
Ross Wightman
3aebc2f06c Switch DPN to use BnAct layer, train a new DPN 68b model with RA to 79.21 2020-07-12 11:13:06 -07:00
Ross Wightman
17f4dd2778 Merge branch 'michalwols-docs' v0.1-rexnet 2020-07-10 12:39:20 -07:00
Ross Wightman
022130d229 Merge branch 'docs' of https://github.com/michalwols/pytorch-image-models into michalwols-docs 2020-07-10 12:05:59 -07:00
Ross Wightman
31cf1251c1
Update setup.py 
Exclude results from possible packaging as it has .py now
 2020-07-09 09:42:36 -07:00
Ross Wightman
0d5550c1e2
Merge pull request #183 from KushajveerSingh/results_diff 
Add rank difference, and top-1/top-5 difference from ImageNet-1k validation for the 3 additional test sets
 2020-07-09 09:41:38 -07:00
KushajveerSingh
5699e8429f set precision=2 in param_count 2020-07-09 16:43:11 +05:30
KushajveerSingh
6526ccb4a7 set precision=3 in results 2020-07-09 16:38:49 +05:30
KushajveerSingh
856f6181f7 add {rank,top1,top5}_diff in results 2020-07-09 03:48:26 +05:30
Ross Wightman
f122f0274b Significant ResNet refactor: 
* stage creation + make_layer moved to separate fn with more sensible dilation/output_stride calc
* drop path rate decay easy to impl with refactored block creation loops
* fix dilation + blur pool combo
 2020-07-05 00:48:12 -07:00
michal
2608918e8d configure mkdocs based documentation 2020-07-03 19:54:17 -04:00
Ross Wightman
a66df5fb91 More model feature extraction support, start to deprecate senet.py, dilations added to regnet, add proper aligned xception 2020-07-03 00:41:30 -07:00
Ross Wightman
7729f40dca Fix another bug, update all gluon resnet models to use new creation method (feature support) 2020-06-30 16:44:15 -07:00
Ross Wightman
d72ac0db25 Fix #173, lr cycle default 0 vs 1. Fix #177, mirror resnest weights for future stability. 2020-06-29 21:31:48 -07:00
Ross Wightman
d0113f9cdb Fix a few issues that came up in tests 2020-06-29 21:13:21 -07:00
Ross Wightman
d23a2697d0 Working on feature extraction, interfaces refined, a number of models working, some in progress. 2020-06-29 18:18:59 -07:00
Ross Wightman
24e7535278 Remove tests from distrib v0.1-resnest 2020-06-16 22:57:26 -07:00
Ross Wightman
328339a086 Add ESE-VoVNet-19-DW weights 2020-06-15 14:31:55 -07:00
Ross Wightman
2d83752bc2 Fix default interpolation/crop of largest 2 ResNeSt models 2020-06-12 16:42:47 -07:00
Ross Wightman
f225ae8e59 Update README with model results and attribution. Make scheduler factory bit more robust to arg differences, add noise to plateau lr and fix min/max. 2020-06-12 13:33:12 -07:00
Ross Wightman
d1b5dddad1
Merge pull request #155 from rwightman/densenet_update_and_more 
DenseNet updates, EvoNorms,  VovNet, activation factory and more. Includes PR #142
 2020-06-11 17:01:19 -07:00
Ross Wightman
39f27c1add Almost ready to merge, vovnet typo, version bump, readme addition 2020-06-11 15:41:58 -07:00
Ross Wightman
151679c2f1 Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory 2020-06-11 14:49:23 -07:00
Ross Wightman
6c7932fe75 Update sotabench.py, tweak VovNet cfg 2020-06-11 14:34:25 -07:00
Ross Wightman
d3ee3de96a Update validation script first batch prime and clear cuda cache between multi-model runs 2020-06-11 13:34:21 -07:00
Ross Wightman
0aca08384f Update regnet cfg keys to match model names so registry works properly 2020-06-09 14:37:45 -07:00
Ross Wightman
b41b8d0108 Update results csv files 2020-06-09 14:36:24 -07:00
Antoine Broyelle
78fa0772cc Leverage python hierachical logger 
with this update one can tune the kind of logs generated by timm but
training and inference traces are unchanged
 2020-06-09 18:28:48 +01:00
Ross Wightman
13c0a6290e Add ESE-VovNet39 weights, 79.32 top-1 2020-06-06 22:50:21 -07:00
Ross Wightman
4ddde1d3a4 Fix two regressions 2020-06-05 11:04:51 -07:00