Commit Graph

348 Commits (ade1ba5fe3782cdb15bd1ff0379f1daaa0f9350b)
 

Author SHA1 Message Date
Ross Wightman ade1ba5fe3 Bring in code that should fix 1d circular padding properly, tweaks to ECA impl, using CECA in MobileNetV3 experiment
5 years ago
Ross Wightman 67e759f710 Adding support to EfficientNet / MobileNetV3 to for different attention layers in .se position
5 years ago
Ross Wightman 27b3680d49 Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.
5 years ago
Ross Wightman 514b0938c4 Experimenting with per-epoch learning rate noise w/ step scheduler
5 years ago
Ross Wightman d77f45a6f6 Forgot to add skresnet34 to sotabench
5 years ago
Ross Wightman 6620770665 Simpler approach to loading entrypoints in hubconf works properly
5 years ago
Ross Wightman e0685dd415
Merge pull request #88 from rwightman/attention
5 years ago
Ross Wightman f1860ef3a5 Update README.md, check in hubconf experiment (fully dynamic)
5 years ago
Ross Wightman 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.
5 years ago
Ross Wightman f1d5f8a6c4 Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights
5 years ago
Ross Wightman 569419b38d Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
5 years ago
Ross Wightman 91e2b33d72 Update results with latest NoisyStudent models and SelectiveKernel models
5 years ago
Ross Wightman 53c47479c4 Batch validation batch size adjustment, tweak L2 crop pct
5 years ago
Ross Wightman 08553e16b3 Merge branch 'master' into attention
5 years ago
Ross Wightman fa38f24967 Update SK network configs, add weights for skresnet8 and skresnext50
5 years ago
Ross Wightman f098fda2ca Add map_location='cpu' to ModelEma resume, should improve #72
5 years ago
Ross Wightman b9496995d8 Add L2-475 PyTorch preprocessing result, update sotabench for new weights
5 years ago
Ross Wightman ba15ca47e8 Add ported EfficientNet-L2, B0-B7 NoisyStudent weights from TF TPU
5 years ago
Ross Wightman 5e6dbbaf30 Add CBAM for experimentation
5 years ago
Ross Wightman d725991870 Remove debug print from ECA module
5 years ago
Ross Wightman 2a7d256fd5 Re-enable mem-efficient/jit activations after torchscript tests
5 years ago
Ross Wightman f902bcd54c Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
5 years ago
Ross Wightman a99ec4e7d1 A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility.
5 years ago
Ross Wightman 13746a33fc Big move, layer modules and fn to timm/models/layers
5 years ago
Ross Wightman f54612f648 Merge branch 'select_kernel' into attention
5 years ago
Ross Wightman 4defbbbaa8 Fix module name mistake, start layers sub-package
5 years ago
Ross Wightman 7011cd0902 A little bit of ECA cleanup
5 years ago
Ross Wightman 46471df7b2 Merge pull request #82 from VRandme/eca
5 years ago
Ross Wightman d0eb59ef46 Remove unused default_init for EfficientNets, experimenting with fanout calc for #84
5 years ago
Ross Wightman cade829105 Add EfficientNet-ES to sotabench
5 years ago
Chris Ha e6a762346a Implement Adaptive Kernel selection
5 years ago
Ross Wightman 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.
5 years ago
Chris Ha 6db087a1ff Merge remote-tracking branch 'upstream/master' into eca
5 years ago
Chris Ha 904c618040 Update EcaModule.py
5 years ago
Chris Ha db91ba053b EcaModule(CamelCase)
5 years ago
Ross Wightman 5eb0e363a6
Update README.md
5 years ago
Ross Wightman 5c4991a088 Add PyTorch trained EfficientNet-ES weights from Andrew Lavin
5 years ago
Chris Ha d04ff95eda Merge branch 'master' into eca
5 years ago
Chris Ha d63ae121d5 Clean up eca_module code
5 years ago
Chris Ha b0d45fd09c Merge remote-tracking branch 'upstream/master'
5 years ago
Ross Wightman d66819d1f3 Indentation mistake. Fixes #81
5 years ago
Ross Wightman 65cda1c63f
Merge pull request #83 from andravin/validation-batch-size-multiplier
5 years ago
Chris Ha f87fcd7e88 Implement Eca modules
5 years ago
Chris Ha 697e05cb3e Merge remote-tracking branch 'upstream/master'
5 years ago
Andrew Lavin b72013def8 Added commandline argument validation-batch-size-multiplier with default set to 1.
5 years ago
Ross Wightman 4808b3c32f Bump version for PyPi update, fix few out of date README items/mistakes, add README updates for TF EfficientNet-B8 (RandAugment)
5 years ago
Ross Wightman 5c85389e26
Update README.md
5 years ago
Ross Wightman 820b73d280
Update README.md
5 years ago
Ross Wightman 82c0a2ff12
Update README.md
5 years ago
Ross Wightman 1ffd2d0694
Add results/README.md
5 years ago