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
387 Commits 141 Branches 65 Tags
Commit Graph

387 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
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
v0.1-tresnet 		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
8a63c1add8 finalizing 2020-04-12 19:08:35 +03:00
talrid
6209146738 TResNet models 2020-04-12 18:44:12 +03:00
mrT23
98a7403ed4
Merge pull request #1 from rwightman/master 
merge master
 2020-04-12 08:44:42 +03:00
Ross Wightman
13cf68850b Remove poorly named metrics from torch imagenet example origins. Use top1/top5 in csv output for consistency with existing validation results files, acc elsewhere. Fixes #111 2020-04-10 14:41:08 -07:00
Ross Wightman
56608c9070 Bump version for pypi 2020-04-09 01:46:05 -07:00
Ross Wightman
0ebcd2eca1 No default img size for inference script. Fixes #109 2020-04-09 01:43:49 -07:00
Ross Wightman
1a8f5900ab Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well. 2020-04-09 01:41:54 -07:00
Ross Wightman
e01ccb88ce
Merge pull request #117 from VRandme/typo_eca 
minor PR to fix typos found in #116 and resnet.py
 2020-04-07 09:36:31 -07:00
Chris Ha
06a50a94a8 Fix minor typos in create_attn.py and resnet.py 
'eca'->'ceca'
and
doest not-> does not
 2020-04-07 21:15:57 +09:00
Ross Wightman
c99a5abed4
Merge pull request #115 from rwightman/mobilenetv2-experiment 
MobileNet-V2 experiments
 2020-04-05 17:32:06 -07:00
Ross Wightman
e34074b4da Add final weights for MobileNet-V2 experiments 2020-04-05 17:31:02 -07:00
Ross Wightman
a6a5565de7 Fiddling... 2020-03-19 10:20:20 -07:00
Ross Wightman
5a16c533ff Add better resnext50_32x4d weights trained by andravin 2020-03-18 14:43:50 -07:00
Ross Wightman
bc998cad91 Experimenting with some MobileNetV2 variations to compare against EfficientNet-Lite 2020-03-18 13:54:06 -07:00
Ross Wightman
71b5cd67da
Merge pull request #105 from rwightman/efficientnet-lite 
EfficientNet-Lite
 2020-03-18 13:18:27 -07:00
Ross Wightman
ef30e7f2b6 Update sotabench with EffNet Lite models 2020-03-18 13:15:54 -07:00
Ross Wightman
3406e582cf Add EfficientNet-Lite results, update README 2020-03-18 13:12:30 -07:00
Ross Wightman
bd05258f7b EfficientNet-Lite model added w/ converted checkpoints, validation in progress... 2020-03-17 23:31:45 -07:00
Ross Wightman
7deacf5477
Merge pull request #99 from andravin/save-last 
Modified save_checkpoint to always save last checkpoint. Fixes #98.
 2020-03-15 11:28:50 -07:00
Andrew Lavin
d92cc4da54 Modified save_checkpoint to always save last checkpoint. Fixes #98. 2020-03-09 22:41:33 -07:00
Ross Wightman
56e2ac3a6d
Merge pull request #94 from rwightman/lr_noise 
Learning rate noise, MobileNetV3 weights, and activate MobileNetV3/EfficientNet weight init change
 2020-02-29 20:41:05 -08:00
Ross Wightman
c16f25ced2 Add MobileNetV3 Large weights, results, update README and sotabench for merge 2020-02-29 20:37:20 -08:00
Ross Wightman
c60069c1eb Annotate types on drop fns to avoid torchscript error 2020-02-27 09:30:23 -08:00
Ross Wightman
9fee316752 Enable fixed fanout calc in EfficientNet/MobileNetV3 weight init by default. Fix #84 2020-02-24 15:11:26 -08:00
Ross Wightman
27b3680d49 Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option. 2020-02-22 16:23:15 -08:00
Ross Wightman
cc5a11abba version bump for PyPi update 2020-02-22 10:26:19 -08:00
Ross Wightman
514b0938c4 Experimenting with per-epoch learning rate noise w/ step scheduler 2020-02-21 11:51:05 -08:00
Ross Wightman
d77f45a6f6 Forgot to add skresnet34 to sotabench 2020-02-18 15:11:41 -08:00
Ross Wightman
6620770665 Simpler approach to loading entrypoints in hubconf works properly 2020-02-18 14:58:23 -08:00
Ross Wightman
e0685dd415
Merge pull request #88 from rwightman/attention 
A lot of attention and much more
 2020-02-18 14:46:49 -08:00
Ross Wightman
f1860ef3a5 Update README.md, check in hubconf experiment (fully dynamic) 2020-02-18 14:41:14 -08:00
Ross Wightman
43225d110c Unify drop connect vs drop path under 'drop path' name, switch all EfficientNet/MobilenetV3 refs to 'drop_path'. Update factory to handle new drop args. 2020-02-18 14:00:26 -08:00
Ross Wightman
f1d5f8a6c4 Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights 2020-02-18 13:58:30 -08:00
Ross Wightman
569419b38d Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch 2020-02-15 21:18:25 -08:00