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

547 Commits (b423bc8362ddac134ad62e331d5bf7652896bcaf)
 

Author SHA1 Message Date
Ross Wightman 5966654052 Update requirements so PyTorch 1.4 is min, add separate sotabench reqs to force 1.4 and prevent error with 1.5 2020-05-24 13:28:03 -07:00
Ross Wightman 0ea53cecc3 Merge branch 'master' into densenet_update_and_more 2020-05-22 16:18:10 -07:00
Ross Wightman 6441e9cc1b Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args. 2020-05-22 16:16:45 -07:00
Ross Wightman d79ac48626 Update sotabench.py 2020-05-22 14:42:43 -07:00
Ross Wightman e881383b4e
Merge pull request #154 from rwightman/tests_bugfixes 
Add backward and default_cfg tests and fix a few issues found. Fix #153
 2020-05-21 19:44:12 -07:00
Ross Wightman 4d13db538f Add x48d ResNext101s to test exclude for ubuntu 2020-05-21 19:13:41 -07:00
Ross Wightman 4212cd3b9f Another attempt at getting Ubuntu test runner to work 2020-05-21 18:55:10 -07:00
Ross Wightman 20329f2630 Bring down test resolutions to see if we can at least do a fwd on the L2 models 2020-05-21 16:49:46 -07:00
Ross Wightman 3873ea710e Minor test change 2020-05-21 15:51:47 -07:00
Ross Wightman afb6bd0669 Add backward and default_cfg tests and fix a few issues found. Fix #153 2020-05-21 15:30:58 -07:00
Ross Wightman ea2e59ca36
Merge pull request #150 from rwightman/regnet 
Add RegNet models and weights
 2020-05-18 00:26:40 -07:00
Ross Wightman 50658b9a67 Add RegNet models and weights 2020-05-18 00:08:52 -07:00
Ross Wightman 7df83258c9 Merge branch 'master' into densenet_update_and_more 2020-05-13 23:34:44 -07:00
Ross Wightman dab9935b36
Merge pull request #148 from rwightman/drop_block_improve 
Improve dropblock impl, add fast variant, better AMP speed, inplace…
 2020-05-13 16:30:34 -07:00
Ross Wightman 1904ed8fec Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups 2020-05-13 15:17:08 -07:00
Ross Wightman 63addb741f
Merge pull request #146 from rwightman/inceptionv3_fix 
Remove annoying torchvision InceptionV3 dependency on scipy and insanely slow tru…
 2020-05-12 22:03:37 -07:00
Ross Wightman 17270c69b9 Remove annoying InceptionV3 dependency on scipy and insanely slow trunc_norm init. Bring InceptionV3 code into this codebase and use upcoming torch trunch_norm_ init. 2020-05-12 21:59:34 -07:00
Ross Wightman c4ca016656
Merge pull request #145 from rwightman/resnest 
ResNeSt
 2020-05-12 13:38:29 -07:00
Ross Wightman 208e7912f7 Missed one of the abalation model entrypoints, update README 2020-05-12 13:36:31 -07:00
Ross Wightman 5bd1ad13e7 Refactor test indent 2020-05-12 13:07:46 -07:00
Ross Wightman 9cc289f18c Exclude EfficientNet-L2 models from test 2020-05-12 13:07:03 -07:00
Ross Wightman 2f884a0ce5 Add resnest14, resnest26, and two of the abalation grouped resnest50 models 2020-05-12 12:21:52 -07:00
Ross Wightman f4cdc2ac31 Add ResNeSt models 2020-05-11 23:27:09 -07:00
Ross Wightman 780860d140 Add norm_act factory method, move JIT of norm layers to factory 2020-05-09 22:09:21 -07:00
Ross Wightman 14edacdf9a DenseNet converted to support ABN (norm + act) modules. Experimenting with EvoNorm, IABN 2020-05-09 18:26:41 -07:00
Ross Wightman e545bb9401
Update test_inference.py 
Not six min
 2020-05-07 10:15:49 -07:00
Ross Wightman 305a2db705
Update test_inference.py 
Make the timeout 5-min for now, see if we can get a pass...
 2020-05-07 10:14:24 -07:00
Ross Wightman 022ed001f3 Update DenseNet to latest in Torchvision (torchscript compat, checkpointing, proper init). Start adding ehanced configurability, stem options... 2020-05-07 09:57:48 -07:00
Ross Wightman 14e01b878c
Merge pull request #143 from michalwols/master 
Setup Github Action to instantiate and run a forward pass with each registered model.
 2020-05-07 09:47:13 -07:00
michal 8c77f14cae Install cpu version of torch on ubuntu 2020-05-07 01:09:16 -04:00
michal 69d725c9fe Basic forward pass test for all registered models 2020-05-07 00:20:58 -04:00
michal 8da43e0617 Install extra dependencies required by some models and log test durations 2020-05-07 00:20:34 -04:00
Mike 16f1f77b41
Add a test workflow for github actions 2020-05-06 23:21:50 -04:00
Vyacheslav Shults a7ebe09029 Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given. 
Make small code refactoring
 2020-05-06 09:54:03 +03:00
Ross Wightman 6cc11a8821
Merge pull request #141 from Animatory/fix_HRNet 
Fixed HRNet modules
 2020-05-05 18:21:34 -07:00
Vyacheslav Shults f0eb021620 Replace all None by nn.Identity() in HRNet modules 2020-05-05 21:09:35 +03:00
Ross Wightman 3b72ebff51
Merge pull request #140 from yoniaflalo/PR_MultiEpochsDataLoader 
added MultiEpochsDataLoader
 2020-05-05 08:51:33 -07:00
AFLALO, Jonathan Isaac a7f570c9b7 added MultiEpochsDataLoader 2020-05-05 14:47:02 +03:00
Ross Wightman 8d8677e03b Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too. 2020-05-04 12:14:43 -07:00
Ross Wightman 353a79aeba
Update README.md 2020-05-03 21:24:49 -07:00
Ross Wightman c9b6f415ca Bump version for Pypi release 2020-05-03 21:18:47 -07:00
Ross Wightman 31f4c126ec Add EfficientNet pruned models to results files 2020-05-03 20:15:11 -07:00
Ross Wightman 375f3e5321 Fix pruned txt files not being installed during pip install 2020-05-03 19:06:14 -07:00
Ross Wightman 8ec554b82e
Merge pull request #136 from yoniaflalo/adding_effnet_pruned 
added efficientnet pruned weights
 2020-05-03 07:14:14 -07:00
AFLALO, Jonathan Isaac 9c15d57505 added efficientnet pruned weights 2020-05-03 09:29:45 +03:00
Ross Wightman a4d20a1cb8
Update README.md 2020-05-01 19:22:35 -07:00
Ross Wightman ea300709f0 Fix model create fn not passing num_classes through. Fix #135 2020-05-01 18:37:03 -07:00
Ross Wightman 779cb0fcc0 Update results with new models 2020-05-01 17:29:29 -07:00
Ross Wightman 2c438c4e0d Merge branch 'master' of github.com:rwightman/pytorch-models 2020-05-01 17:03:17 -07:00
Ross Wightman 28739bb721 Merge branch 'VRandme-blur' 2020-05-01 17:01:43 -07:00