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

662 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Ross Wightman
5d4c3d0af3 Add enhanced ParserImageInTar that can read images from tars within tars, folders with multiple tars, etc. Additional comment cleanup. 2021-01-22 10:52:04 -08:00
Ross Wightman
55f7dfa9ea Refactor vision_transformer entrpy fns, add pos embedding resize support for fine tuning, add some deit models for testing 2021-01-18 16:11:02 -08:00
Ross Wightman
9d5d4b8df6 Fix silly train.py typo during dataset work 2021-01-16 16:32:21 -08:00
Ross Wightman
d55bcc0fee Finishing adding stochastic depth support to BiT ResNetV2 models 2021-01-16 16:32:03 -08:00
Ross Wightman
0a1668f63e Update tests 2021-01-16 12:37:10 -08:00
Ross Wightman
58ccf43150 Add BiT references and knowledge distill links to readme/docs 2021-01-16 12:30:02 -08:00
Ross Wightman
855d6cc217 More dataset work including factories and a tensorflow datasets (TFDS) wrapper 
* Add parser/dataset factory methods for more flexible dataset & parser creation
* Add dataset parser that wraps TFDS image classification datasets
* Tweak num_classes handling bug for 21k models
* Add initial deit models so they can be benchmarked in next csv results runs
 2021-01-15 17:26:20 -08:00
Ross Wightman
20516abc18 Fix some broken tests for ResNetV2 BiT models 2021-01-04 23:21:39 -08:00
Ross Wightman
fd9061dbf7 Remove debug print from train.py 2021-01-04 12:16:06 -08:00
Ross Wightman
59ec7e6a53 Merge branch 'master' into imagenet21k_datasets_more 2021-01-04 12:11:05 -08:00
Ross Wightman
fc3d9183e8
Merge pull request #335 from kecsap/new_option 
Add --input-size option to scripts to specify full input dimensions f…
 2021-01-04 10:32:21 -08:00
Ross Wightman
e7a9ddf982
Merge pull request #334 from kecsap/links 
Follow symbolic links during dataset scanning
 2021-01-04 10:30:58 -08:00
Ross Wightman
19816fe226 Add citation info 2021-01-03 15:46:12 -08:00
Csaba Kertesz
e42b140ade Add --input-size option to scripts to specify full input dimensions from command-line 2021-01-04 00:25:29 +02:00
Csaba Kertesz
7cae7e7035 Follow links during dataset scanning 2021-01-04 00:16:45 +02:00
Ross Wightman
1d01c2b68c
Update README.md v0.3.3 2021-01-03 12:49:31 -08:00
Ross Wightman
c96e9f99a0 Update version to 0.3.3 2021-01-03 12:43:44 -08:00
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
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