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

729 Commits (82c1fe958a2a110827896f499f49337ec9fbd6ad)
 

Author SHA1 Message Date
AFLALO, Jonathan Isaac 82c1fe958a added hardcorenas model 
refactoring to support feature extraction
 2021-03-15 08:06:34 +02:00
Ross Wightman 5c7d298234 Update README.md, documentation, add TOC, sponsors, and links to timmdocs and paperswithcode 2021-03-10 10:24:23 -08:00
Ross Wightman f1f332fee5
Update README.md 2021-03-07 16:43:52 -08:00
Ross Wightman 5b28ef4100
Merge pull request #475 from rwightman/pre-release/0.4.5 
Prep for PyPi release. Tweak NFNet, ResNetV2, RexNet feature extraction, use pre-act features. Also, add act args to RexNet #202
 2021-03-07 16:18:31 -08:00
Ross Wightman f57db99101 Update README, fix iabn pip version print. 2021-03-07 16:17:06 -08:00
Ross Wightman 7ef7788ee9 Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk() 2021-03-06 16:56:34 -08:00
Ross Wightman 317ea3e599 Tweak NFNet, ResNetV2, RexNet feature extraction, use pre-act features for simplicty. Update tests for pytorch 1.8. 0.4.5 version bump. 2021-03-06 12:48:06 -08:00
Ross Wightman 5fcdd4b822
Merge pull request #459 from onatyap/patch-2 
corrected typo in feature_extraction.md
 2021-03-05 10:27:47 -08:00
Ross Wightman 95feb1da41 Update results csvs, run with PyTorch 1.7.1 cuda 10.2, some differences from prev run with older PT ver. 2021-02-28 17:43:22 -08:00
Onat Yapici 3b95fa4248
corrected typo in feature_extraction.md 
severay changed to several
 2021-02-28 16:25:00 +03:00
Ross Wightman 48371a33b1
Create FUNDING.yml 
$cale up the training...
 2021-02-18 16:12:53 -08:00
Ross Wightman da4839530c Fix test model filter to include dm_ variants that break GitHub CI limits 2021-02-18 13:43:04 -08:00
Ross Wightman 8563609b28 Update notes in ScaledStdConv impl 2021-02-18 12:44:08 -08:00
Ross Wightman 678ba4e0a2 Add NFNet-F model weights ported from DeepMind Haiku impl and new set of models w/ compatible config. 2021-02-18 12:28:46 -08:00
Ross Wightman 4ea5931964
Merge pull request #437 from rwightman/agc 
Adaptive Gradient Clipping (AGC) Impl
 2021-02-16 10:27:58 -08:00
Ross Wightman 361fd0fc40
Update README.md 2021-02-16 10:27:41 -08:00
Ross Wightman 9de2ec5e44 Update README for AGC and bump version to 0.4.4 2021-02-16 09:13:03 -08:00
Ross Wightman 01653db104 Missed clip-mode arg for repo train script 2021-02-15 23:27:16 -08:00
Ross Wightman 4f49b94311 Initial AGC impl. Still testing. 2021-02-15 23:22:44 -08:00
Ross Wightman 5f9aff395c Fix stem width in NFNet-F models, add some more comments, add some 'light' NFNet models for testing. 2021-02-13 16:58:51 -08:00
Ross Wightman 4df513c68f
Merge pull request #427 from rwightman/nfnet 
Add NFNet-F models and tweak existing NF models.
 2021-02-12 22:08:03 -08:00
Ross Wightman d86dbe45c2 Update README.md and few more comments 2021-02-12 22:07:18 -08:00
Ross Wightman 0d253e2c5e Fix issue with nfnet tests, bit more cleanup. 2021-02-12 21:05:41 -08:00
Ross Wightman cb06c7a910 Add NFNet-F models and tweak existing NF models. 2021-02-12 18:28:56 -08:00
Ross Wightman 607f9149b1
Update README.md 2021-02-11 22:06:06 -08:00
Ross Wightman e4de077021 Add first 'Normalizer Free' models. nf_regnet_b1 79.3 @ 288x288 test, and nf_resnet50 80.3 @ 256x256 test (80.68 @ 288x288). 2021-02-11 13:20:11 -08:00
Ross Wightman d8e69206be
Merge pull request #419 from rwightman/byob_vgg_models 
More models, GPU-Efficient Nets, RepVGG, classic VGG, and flexible Byob backbone.
 2021-02-10 15:44:09 -08:00
Ross Wightman ca9b078ac7 Update README.md and docs. Version bumped to 0.4.3 2021-02-10 14:46:07 -08:00
Ross Wightman 6853b07bbd Improve RegVGG block identity/vs non for clariy and fix attn usage. Add comments. 2021-02-10 14:40:29 -08:00
Ross Wightman 0356e773f5 Default to native PyTorch AMP instead of APEX amp. Too many APEX issues cropping up lately. 2021-02-10 14:31:18 -08:00
Ross Wightman b9bd960a03
Merge pull request #421 from aboutmedicine/master 
Update collections.abc import to avoid pytorch 1.8 issue. Fix #420
 2021-02-10 08:11:49 -08:00
Reuben 94ca140b67 update collections.abc import 2021-02-10 23:54:35 +11:00
Ross Wightman db6128db0f
Merge pull request #418 from kecsap/master 
Change the Python interpreter to Python 3.x in the scripts
 2021-02-09 20:55:55 -08:00
Ross Wightman b4e216e377 Fix a few small things. 2021-02-09 17:33:43 -08:00
Ross Wightman dc85e5a237 Add ByobNet w/ GPU-EfficientNets and RepVGG. Also add classic vgg models. 2021-02-09 16:22:52 -08:00
Ross Wightman 1bcc69e0ad Use in_channels for depthwise groups, allows using `out_channels=N * in_channels` (does not impact existing models). Fix #354. 2021-02-09 16:22:52 -08:00
Ross Wightman 9811e229f7 Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models. 2021-02-09 16:22:52 -08:00
Csaba Kertesz 5114c214fc Change the Python interpreter to Python 3.x in the scripts 2021-02-09 21:20:28 +02:00
Ross Wightman aaa715b1e9
Update README.md 2021-02-08 12:04:48 -08:00
Ross Wightman a7f95818e4
Merge pull request #413 from rwightman/eca-weights 
Add new weights for ecaresnet26t/50t/269d models.
 2021-02-08 12:03:19 -08:00
Ross Wightman a39c3ee216
Merge branch 'master' into eca-weights 2021-02-08 11:52:31 -08:00
Ross Wightman e9d6fe293c Update README for new weights. Version 0.4.2 2021-02-08 11:51:16 -08:00
Ross Wightman 666de85cf1 Move stride in EdgeResidual block to 3x3 expansion conv. Fix #414 2021-02-07 22:10:18 -08:00
Ross Wightman 3b57490a63 Fix some half removed resnet model defs, pooling for ecaresnet269d 2021-02-07 22:09:25 -08:00
Ross Wightman 2a8c4dc63b Add validation script update for using test_input_size in model default_cfgs 2021-02-07 21:35:50 -08:00
Ross Wightman 68a4144882 Add new weights for ecaresnet26t/50t/269d models. Remove distinction between 't' and 'tn' (tiered models), tn is now t. Add test time img size spec to default cfg. 2021-02-06 16:30:02 -08:00
Ross Wightman 3a7aa95f7e
Update README.md 2021-02-05 07:39:21 -08:00
Ross Wightman b9843f954b
Merge pull request #282 from tigert1998/patch-1 
Add symbolic for SwishJitAutoFn to support onnx
 2021-02-04 12:18:40 -08:00
Ross Wightman ea36a78cff
Merge pull request #401 from hwangdeyu/deyu/add_HardSwishJitAutoFn_operator 
add HardSwishJitAutoFn operator export to onnx
 2021-02-04 12:18:19 -08:00
hwangdeyu 7a4be5c035 add operator HardSwishJitAutoFn export to onnx 2021-02-03 09:06:53 +08:00