6cc11a8821
Merge pull request #141 from Animatory/fix_HRNet
...
Fixed HRNet modules
2020-05-05 18:21:34 -07:00
f0eb021620
Replace all None by nn.Identity() in HRNet modules
2020-05-05 21:09:35 +03:00
3b72ebff51
Merge pull request #140 from yoniaflalo/PR_MultiEpochsDataLoader
...
added MultiEpochsDataLoader
2020-05-05 08:51:33 -07:00
a7f570c9b7
added MultiEpochsDataLoader
2020-05-05 14:47:02 +03:00
8d8677e03b
Fix #139 . Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.
2020-05-04 12:14:43 -07:00
353a79aeba
Update README.md
2020-05-03 21:24:49 -07:00
c9b6f415ca
Bump version for Pypi release
2020-05-03 21:18:47 -07:00
31f4c126ec
Add EfficientNet pruned models to results files
2020-05-03 20:15:11 -07:00
375f3e5321
Fix pruned txt files not being installed during pip install
2020-05-03 19:06:14 -07:00
8ec554b82e
Merge pull request #136 from yoniaflalo/adding_effnet_pruned
...
added efficientnet pruned weights
2020-05-03 07:14:14 -07:00
9c15d57505
added efficientnet pruned weights
2020-05-03 09:29:45 +03:00
a4d20a1cb8
Update README.md
2020-05-01 19:22:35 -07:00
ea300709f0
Fix model create fn not passing num_classes through. Fix #135
2020-05-01 18:37:03 -07:00
779cb0fcc0
Update results with new models
2020-05-01 17:29:29 -07:00
2c438c4e0d
Merge branch 'master' of github.com:rwightman/pytorch-models
2020-05-01 17:03:17 -07:00
28739bb721
Merge branch 'VRandme-blur'
2020-05-01 17:01:43 -07:00
2681a8d618
Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop
2020-05-01 17:00:21 -07:00
9590f301a9
Merge branch 'blur' of https://github.com/VRandme/pytorch-image-models into VRandme-blur
2020-05-01 15:19:57 -07:00
a5220ad3aa
Merge pull request #125 from Separius/patch-1
...
fix typo in eca
2020-05-01 15:11:56 -07:00
7a9942a75e
Merge branch 'yoniaflalo-adding_ECA_resnet'
2020-05-01 15:10:23 -07:00
0834fbc01c
Move pruned model adapt strings to separate txt files. A few minor formatting alignment tweaks.
2020-05-01 15:09:20 -07:00
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
07f19dd699
added eca resnet
2020-04-29 19:09:55 +03:00
1d4ac1b165
Bump version for pypi release. Fix #130
2020-04-27 18:02:40 -07:00
be7c784d21
Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference
2020-04-27 17:50:19 -07:00
e3a98171b2
Remove some CRLF that made it into TResNet merge
2020-04-27 13:23:41 -07:00
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
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
02a30411ad
Replace fp16 with amp support for validate.py script
2020-04-27 12:31:36 -07:00
e6f24e5578
Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults
2020-04-25 19:42:13 -07:00
20290b5b56
fix typo in eca
2020-04-14 15:20:35 +04:30
ebf82b84ac
Merge pull request #122 from mrT23/master
...
TResNet models
2020-04-12 18:23:46 -07:00
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
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
8a63c1add8
finalizing
2020-04-12 19:08:35 +03:00
6209146738
TResNet models
2020-04-12 18:44:12 +03:00
98a7403ed4
Merge pull request #1 from rwightman/master
...
merge master
2020-04-12 08:44:42 +03:00
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
56608c9070
Bump version for pypi
2020-04-09 01:46:05 -07:00
0ebcd2eca1
No default img size for inference script. Fixes #109
2020-04-09 01:43:49 -07:00
1a8f5900ab
Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
2020-04-09 01:41:54 -07:00
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
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
1a9ab07307
One too many changes at a time, fix missing C
2020-04-07 21:12:08 +09:00
f17b42bc33
Blur filter no longer a buffer
2020-04-07 21:12:08 +09:00
6cdeca24a3
Some cleanup and fixes for initial BlurPool impl. Still some testing and tweaks to go...
2020-04-07 21:12:08 +09:00
acd1b6cccd
Implement Functional Blur on resnet.py
...
1. add ResNet argument blur=''
2. implement blur for maxpool and strided convs in downsampling blocks
2020-04-07 21:12:08 +09:00
ce3d82b58b
Update blurpool.py
...
clean up code for PR
2020-04-07 21:12:07 +09:00
3a287a6e76
Create blurpool.py
...
Initial implementation of blur layer.
currently tests as correct against Downsample of original github
2020-04-07 21:12:07 +09:00
c99a5abed4
Merge pull request #115 from rwightman/mobilenetv2-experiment
...
MobileNet-V2 experiments
2020-04-05 17:32:06 -07:00
Ross Wightman
Vyacheslav Shults
AFLALO, Jonathan Isaac
Ross Wightman
Sepehr Sameni
Alexey Chernov
talrid
mrT23
Chris Ha