Commit Graph

518 Commits (b1f1a54de92ef8629eb486f2b14035a8cb314766)
 

Author SHA1 Message Date
Ross Wightman 1a9ab07307 One too many changes at a time, fix missing C
4 years ago
Ross Wightman f17b42bc33 Blur filter no longer a buffer
4 years ago
Ross Wightman 6cdeca24a3 Some cleanup and fixes for initial BlurPool impl. Still some testing and tweaks to go...
4 years ago
Chris Ha acd1b6cccd Implement Functional Blur on resnet.py
4 years ago
Chris Ha ce3d82b58b Update blurpool.py
4 years ago
Chris Ha 3a287a6e76 Create blurpool.py
4 years ago
Ross Wightman c99a5abed4
Merge pull request #115 from rwightman/mobilenetv2-experiment
4 years ago
Ross Wightman e34074b4da Add final weights for MobileNet-V2 experiments
4 years ago
Ross Wightman a6a5565de7 Fiddling...
4 years ago
Ross Wightman 5a16c533ff Add better resnext50_32x4d weights trained by andravin
4 years ago
Ross Wightman bc998cad91 Experimenting with some MobileNetV2 variations to compare against EfficientNet-Lite
4 years ago
Ross Wightman 71b5cd67da
Merge pull request #105 from rwightman/efficientnet-lite
4 years ago
Ross Wightman ef30e7f2b6 Update sotabench with EffNet Lite models
4 years ago
Ross Wightman 3406e582cf Add EfficientNet-Lite results, update README
4 years ago
Ross Wightman bd05258f7b EfficientNet-Lite model added w/ converted checkpoints, validation in progress...
4 years ago
Ross Wightman 7deacf5477
Merge pull request #99 from andravin/save-last
4 years ago
Andrew Lavin d92cc4da54 Modified save_checkpoint to always save last checkpoint. Fixes #98.
4 years ago
Ross Wightman 56e2ac3a6d
Merge pull request #94 from rwightman/lr_noise
4 years ago
Ross Wightman c16f25ced2 Add MobileNetV3 Large weights, results, update README and sotabench for merge
4 years ago
Ross Wightman c60069c1eb Annotate types on drop fns to avoid torchscript error
4 years ago
Ross Wightman 9fee316752 Enable fixed fanout calc in EfficientNet/MobileNetV3 weight init by default. Fix #84
4 years ago
Ross Wightman 27b3680d49 Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.
4 years ago
Ross Wightman cc5a11abba version bump for PyPi update
4 years ago
Ross Wightman 514b0938c4 Experimenting with per-epoch learning rate noise w/ step scheduler
4 years ago
Ross Wightman d77f45a6f6 Forgot to add skresnet34 to sotabench
4 years ago
Ross Wightman 6620770665 Simpler approach to loading entrypoints in hubconf works properly
4 years ago
Ross Wightman e0685dd415
Merge pull request #88 from rwightman/attention
4 years ago
Ross Wightman f1860ef3a5 Update README.md, check in hubconf experiment (fully dynamic)
4 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.
4 years ago
Ross Wightman f1d5f8a6c4 Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights
4 years ago
Ross Wightman 569419b38d Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
4 years ago
Ross Wightman 91e2b33d72 Update results with latest NoisyStudent models and SelectiveKernel models
4 years ago
Ross Wightman 53c47479c4 Batch validation batch size adjustment, tweak L2 crop pct
4 years ago
Ross Wightman 08553e16b3 Merge branch 'master' into attention
4 years ago
Ross Wightman fa38f24967 Update SK network configs, add weights for skresnet8 and skresnext50
4 years ago
Ross Wightman f098fda2ca Add map_location='cpu' to ModelEma resume, should improve #72
4 years ago
Ross Wightman b9496995d8 Add L2-475 PyTorch preprocessing result, update sotabench for new weights
4 years ago
Ross Wightman ba15ca47e8 Add ported EfficientNet-L2, B0-B7 NoisyStudent weights from TF TPU
4 years ago
Ross Wightman 5e6dbbaf30 Add CBAM for experimentation
4 years ago
Ross Wightman d725991870 Remove debug print from ECA module
4 years ago
Ross Wightman 2a7d256fd5 Re-enable mem-efficient/jit activations after torchscript tests
4 years ago
Ross Wightman f902bcd54c Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
4 years ago
Ross Wightman a99ec4e7d1 A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility.
4 years ago
Ross Wightman 13746a33fc Big move, layer modules and fn to timm/models/layers
4 years ago
Ross Wightman f54612f648 Merge branch 'select_kernel' into attention
4 years ago
Ross Wightman 4defbbbaa8 Fix module name mistake, start layers sub-package
4 years ago
Ross Wightman 7011cd0902 A little bit of ECA cleanup
4 years ago
Ross Wightman 46471df7b2 Merge pull request #82 from VRandme/eca
4 years ago
Ross Wightman d0eb59ef46 Remove unused default_init for EfficientNets, experimenting with fanout calc for #84
4 years ago
Ross Wightman cade829105 Add EfficientNet-ES to sotabench
4 years ago