Commit Graph

1500 Commits (f1d2160d857c1d98942e16ad3b59d7b74ecb2255)
 

Author SHA1 Message Date
Ross Wightman 9fee316752 Enable fixed fanout calc in EfficientNet/MobileNetV3 weight init by default. Fix #84
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 cc5a11abba version bump for PyPi update
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 670c61b28f Some cutmix/mixup cleanup/fixes
5 years ago
Ross Wightman b3cb5f3275 Working on CutMix impl as per #8, integrating with Mixup, currently experimenting...
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