71b5cd67da
Merge pull request #105 from rwightman/efficientnet-lite
...
EfficientNet-Lite
2020-03-18 13:18:27 -07:00
ef30e7f2b6
Update sotabench with EffNet Lite models
2020-03-18 13:15:54 -07:00
3406e582cf
Add EfficientNet-Lite results, update README
2020-03-18 13:12:30 -07:00
bd05258f7b
EfficientNet-Lite model added w/ converted checkpoints, validation in progress...
2020-03-17 23:31:45 -07:00
7deacf5477
Merge pull request #99 from andravin/save-last
...
Modified save_checkpoint to always save last checkpoint. Fixes #98 .
2020-03-15 11:28:50 -07:00
d92cc4da54
Modified save_checkpoint to always save last checkpoint. Fixes #98 .
2020-03-09 22:41:33 -07:00
56e2ac3a6d
Merge pull request #94 from rwightman/lr_noise
...
Learning rate noise, MobileNetV3 weights, and activate MobileNetV3/EfficientNet weight init change
2020-02-29 20:41:05 -08:00
c16f25ced2
Add MobileNetV3 Large weights, results, update README and sotabench for merge
2020-02-29 20:37:20 -08:00
c60069c1eb
Annotate types on drop fns to avoid torchscript error
2020-02-27 09:30:23 -08:00
9fee316752
Enable fixed fanout calc in EfficientNet/MobileNetV3 weight init by default. Fix #84
2020-02-24 15:11:26 -08:00
27b3680d49
Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.
2020-02-22 16:23:15 -08:00
cc5a11abba
version bump for PyPi update
2020-02-22 10:26:19 -08:00
514b0938c4
Experimenting with per-epoch learning rate noise w/ step scheduler
2020-02-21 11:51:05 -08:00
d77f45a6f6
Forgot to add skresnet34 to sotabench
2020-02-18 15:11:41 -08:00
6620770665
Simpler approach to loading entrypoints in hubconf works properly
2020-02-18 14:58:23 -08:00
e0685dd415
Merge pull request #88 from rwightman/attention
...
A lot of attention and much more
2020-02-18 14:46:49 -08:00
f1860ef3a5
Update README.md, check in hubconf experiment (fully dynamic)
2020-02-18 14:41:14 -08:00
43225d110c
Unify drop connect vs drop path under 'drop path' name, switch all EfficientNet/MobilenetV3 refs to 'drop_path'. Update factory to handle new drop args.
2020-02-18 14:00:26 -08:00
f1d5f8a6c4
Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights
2020-02-18 13:58:30 -08:00
569419b38d
Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
2020-02-15 21:18:25 -08:00
91e2b33d72
Update results with latest NoisyStudent models and SelectiveKernel models
2020-02-15 20:38:57 -08:00
53c47479c4
Batch validation batch size adjustment, tweak L2 crop pct
2020-02-15 20:37:04 -08:00
08553e16b3
Merge branch 'master' into attention
2020-02-14 18:24:21 -08:00
fa38f24967
Update SK network configs, add weights for skresnet8 and skresnext50
2020-02-14 15:37:00 -08:00
f098fda2ca
Add map_location='cpu' to ModelEma resume, should improve #72
2020-02-12 13:23:56 -08:00
b9496995d8
Add L2-475 PyTorch preprocessing result, update sotabench for new weights
2020-02-12 13:20:37 -08:00
ba15ca47e8
Add ported EfficientNet-L2, B0-B7 NoisyStudent weights from TF TPU
2020-02-12 11:26:38 -08:00
5e6dbbaf30
Add CBAM for experimentation
2020-02-10 16:23:09 -08:00
d725991870
Remove debug print from ECA module
2020-02-10 16:21:33 -08:00
2a7d256fd5
Re-enable mem-efficient/jit activations after torchscript tests
2020-02-10 11:59:36 -08:00
f902bcd54c
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
...
* select_conv2d -> create_conv2d
* added create_attn to create attention module from string/bool/module
* factor padding helpers into own file, use in both conv2d_same and avg_pool2d_same
* add some more test eca resnet variants
* minor tweaks, naming, comments, consistency
2020-02-10 11:55:03 -08:00
a99ec4e7d1
A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility.
2020-02-09 14:46:28 -08:00
13746a33fc
Big move, layer modules and fn to timm/models/layers
2020-02-09 13:13:08 -08:00
f54612f648
Merge branch 'select_kernel' into attention
2020-02-09 12:59:24 -08:00
4defbbbaa8
Fix module name mistake, start layers sub-package
2020-02-09 12:44:26 -08:00
7011cd0902
A little bit of ECA cleanup
2020-02-09 12:41:18 -08:00
46471df7b2
Merge pull request #82 from VRandme/eca
...
ECA-Net Efficient Channel Attention
2020-02-09 12:31:05 -08:00
d0eb59ef46
Remove unused default_init for EfficientNets, experimenting with fanout calc for #84
2020-02-09 11:33:32 -08:00
cade829105
Add EfficientNet-ES to sotabench
2020-02-09 11:33:32 -08:00
e6a762346a
Implement Adaptive Kernel selection
...
When channel size is given,
calculate adaptive kernel size according to original paper.
Otherwise use the given kernel size(k_size), which defaults to 3
2020-02-09 11:58:03 +09:00
13e8da2b46
SelectKernel split_input works best when input channels split like grouped conv, but output is full width. Disable zero_init for SK nets, seems a bad combo.
2020-02-07 22:42:04 -08:00
6db087a1ff
Merge remote-tracking branch 'upstream/master' into eca
2020-02-07 19:36:35 +09:00
904c618040
Update EcaModule.py
...
Make pylint happy
(commas, unused imports, missed imports)
2020-02-07 19:36:18 +09:00
db91ba053b
EcaModule(CamelCase)
...
CamelCased EcaModule.
Renamed all instances of ecalayer to EcaModule.
eca_module.py->EcaModule.py
2020-02-07 19:28:07 +09:00
5eb0e363a6
Update README.md
2020-02-06 12:56:46 -08:00
5c4991a088
Add PyTorch trained EfficientNet-ES weights from Andrew Lavin
2020-02-06 12:53:55 -08:00
d04ff95eda
Merge branch 'master' into eca
2020-02-06 22:44:52 +09:00
d63ae121d5
Clean up eca_module code
...
functionally similar
adjusted rwightman's version of reshaping and viewing.
Use F.pad for circular eca version for cleaner code
2020-02-06 22:44:33 +09:00
b0d45fd09c
Merge remote-tracking branch 'upstream/master'
2020-02-06 22:00:37 +09:00
d66819d1f3
Indentation mistake. Fixes #81
2020-02-04 22:56:00 -08:00
Ross Wightman
Ross Wightman
Andrew Lavin
Chris Ha