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

746 Commits (ead80d33c59ca18acaaa594159ca28b6116e9acb)
 

Author SHA1 Message Date
Ross Wightman a7d0a8b5b2 Update results csv files with latest models, incl 101D, 152D, 200D, SE152D ResNets and yet to be merged BiT and ViT-R50 models. 2021-01-03 12:42:43 -08:00
Ross Wightman 4e2533db77 Add 320x320 model default cfgs for 101D and 152D ResNets. Add SEResNet-152D weights and 320x320 cfg. 2021-01-03 12:10:25 -08:00
Ross Wightman 0167f749d3 Remove some old __future__ imports 2021-01-03 11:24:16 -08:00
Ross Wightman 85bf4b8cd6 Add setup.cfg for conda / fastai integration 2021-01-03 11:22:11 -08:00
Ross Wightman e553480b67 Add 21843 synset txt for google 21k models like BiT/ViT 2020-12-28 16:59:15 -08:00
Ross Wightman e35e9760a6 More work on dataset / parser split and imagenet21k (tar) support 2020-12-28 16:59:15 -08:00
Ross Wightman ce69de70d3 Add 21k weight urls to vision_transformer. Cleanup feature_info for preact ResNetV2 (BiT) models 2020-12-28 16:59:15 -08:00
Ross Wightman 231d04e91a ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing. 2020-12-28 16:59:15 -08:00
Ross Wightman de6046e213 Initial commit for dataset / parser reorg to support additional datasets / types 2020-12-28 16:59:15 -08:00
Ross Wightman 392595c7eb Add pool_size to default cfgs for new models to prevent tests from failing. Add explicit 200D_320 model entrypoint for next benchmark run. 2020-12-18 21:28:47 -08:00
Ross Wightman 7a75b8d033
Update README.md 2020-12-18 17:27:57 -08:00
Ross Wightman b1f1228a41 Add ResNet101D, 152D, and 200D weights, remove meh 66d model 2020-12-18 17:13:37 -08:00
Ross Wightman 198f6ea0f3
Merge pull request #302 from Jasha10/create_optimizer-opt_args 
Configure create_optimizer with args.opt_args
 2020-12-08 10:19:36 -08:00
Jasha 7c56c718f3 Configure create_optimizer with args.opt_args 
Closes #301
 2020-12-08 00:03:09 -06:00
Ross Wightman 51d74d91da Update README.md 2020-12-07 09:59:50 -08:00
Ross Wightman 9a25fdf3ad
Merge pull request #297 from rwightman/ema_simplify 
Simplified JIT compatible Ema module. Fixes for SiLU export and torchscript training w/ Linear layer.
 2020-12-05 11:42:45 -08:00
Ross Wightman c9ebe86d03
Merge pull request #300 from tmkkk/real-labels-fix 
Fix a bug with accuracy retrieving from RealLabels
 2020-12-04 08:36:38 -08:00
Tymoteusz Wiśniewski de15b43865 Fix a bug with accuracy retrieving from RealLabels 2020-12-04 16:12:50 +01:00
Ross Wightman cd72e66eff Bug in last mod for features_only default_cfg 2020-12-03 12:33:01 -08:00
Ross Wightman 867a0e5a04 Add default_cfg back to models wrapped in feature extraction module as per discussion in #294. 2020-12-03 10:24:35 -08:00
Ross Wightman 4ca52d73d8 Add separate set and update method to ModelEmaV2 2020-12-03 10:05:09 -08:00
Ross Wightman 2ed8f24715 A few more changes for 0.3.2 maint release. Linear layer change for mobilenetv3 and inception_v3, support no bias for linear wrapper. 2020-11-30 16:19:52 -08:00
Ross Wightman 6504a42832 Version 0.3.2 2020-11-30 13:39:08 -08:00
Ross Wightman 460eba7f24 Work around casting issue with combination of native torch AMP and torchscript for Linear layers 2020-11-30 13:30:51 -08:00
Ross Wightman 5f4b6076d8 Fix inplace arg compat for GELU and PreLU via activation factory 2020-11-30 13:27:40 -08:00
Ross Wightman fd962c4b4a Native SiLU (Swish) op doesn't export to ONNX 2020-11-29 21:56:55 -08:00
Ross Wightman 27bbc70d71 Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training. 2020-11-29 16:22:19 -08:00
Ross Wightman 6f43aeb252
Merge pull request #286 from s-rog/patch-1 
Fix link typo in README
 2020-11-24 09:47:15 -08:00
Roger Shieh a7f6126b92
Update README.md 2020-11-24 09:02:37 +08:00
tigertang 43f2500c26
Add symbolic for SwishJitAutoFn to support onnx 2020-11-18 14:36:12 +08:00
Ross Wightman 9214ca0716 Simplifying EMA... 2020-11-16 12:51:52 -08:00
Ross Wightman 80cd31f21f
Create config.yml 2020-11-12 20:55:27 -08:00
Ross Wightman d940a53cd9 Update issue templates 2020-11-10 12:16:40 -08:00
Ross Wightman 60c998af22 Update issue templates 2020-11-10 11:39:52 -08:00
Ross Wightman 9d73911e62 Update issue templates 2020-11-10 11:35:30 -08:00
Ross Wightman 2542283b28
Merge pull request #268 from seemethere/patch-1 
Update torch -> 1.7.0, torchvision -> 0.8.1 for github actions
 2020-11-02 11:46:21 -08:00
Eli Uriegas ab9f1fc995
Update torch -> 1.7.0, torchvision -> 0.8.1 2020-11-02 10:20:17 -08:00
Ross Wightman 53aeed3499 ver 0.3.1 2020-10-31 18:14:58 -07:00
Ross Wightman 30ab4a1494 Fix issue in optim factory with sgd / eps flag. Bump version to 0.3.1 2020-10-31 18:05:30 -07:00
Ross Wightman 46f15443be Update README.md and docs in prep for 0.3.0 PyPi release. 2020-10-30 13:27:19 -07:00
Ross Wightman af3299ba4a
Merge pull request #263 from rwightman/fixes_oct2020 
Fixes for upcoming PyPi release
 2020-10-29 20:14:13 -07:00
Ross Wightman 741572dc9d Bump version to 0.3.0 for pending PyPi push 2020-10-29 17:31:39 -07:00
Ross Wightman b401952caf Add newly added vision transformer large/base 224x224 weights ported from JAX official repo 2020-10-29 17:31:01 -07:00
Ross Wightman 61200db0ab in_chans=1 working w/ pretrained weights for vision_transformer 2020-10-29 15:49:36 -07:00
Ross Wightman e90edce438 Support native silu activation (aka swish). An optimized ver is available in PyTorch 1.7. 2020-10-29 15:45:17 -07:00
Ross Wightman da6cd2cc1f Fix regression for pretrained classifier loading when using entrypt functions directly 2020-10-29 15:43:39 -07:00
Ross Wightman f591e90b0d Make sure num_features attr is present in vit models as with others 2020-10-29 15:33:47 -07:00
Ross Wightman 4a3df7842a Fix topn metric view regression on PyTorch 1.7 2020-10-29 14:04:15 -07:00
Ross Wightman f944242cb0 Fix #262, num_classes arg mixup. Make vision_transformers a bit closer to other models wrt get/reset classfier/forward_features. Fix torchscript for ViT. 2020-10-29 13:58:28 -07:00
Ross Wightman da1b90e5c9 Update results csvs with latest run 2020-10-29 07:55:13 -07:00