Commit Graph

202 Commits (779cb0fcc001d39805a8398a64222e55b59f0408)

Author SHA1 Message Date
Ross Wightman 2c438c4e0d Merge branch 'master' of github.com:rwightman/pytorch-models
4 years ago
Ross Wightman 2681a8d618 Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop
4 years ago
Ross Wightman 9590f301a9 Merge branch 'blur' of https://github.com/VRandme/pytorch-image-models into VRandme-blur
4 years ago
Ross Wightman a5220ad3aa
Merge pull request #125 from Separius/patch-1
4 years ago
Ross Wightman 0834fbc01c Move pruned model adapt strings to separate txt files. A few minor formatting alignment tweaks.
4 years ago
Ross Wightman ba793f5c1c Merge branch 'adding_ECA_resnet' of https://github.com/yoniaflalo/pytorch-image-models into yoniaflalo-adding_ECA_resnet
4 years ago
AFLALO, Jonathan Isaac 07f19dd699 added eca resnet
4 years ago
Ross Wightman 1d4ac1b165 Bump version for pypi release. Fix #130
4 years ago
Ross Wightman be7c784d21 Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference
4 years ago
Ross Wightman e3a98171b2 Remove some CRLF that made it into TResNet merge
4 years ago
Ross Wightman 0004f37d25 Some TResNet cleanup.
4 years ago
Ross Wightman 64fe37d008 Remove old mean/std helper, rely fully on cmd line or default_cfg now. Fixes #126
4 years ago
Ross Wightman e6f24e5578 Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults
4 years ago
Sepehr Sameni 20290b5b56
fix typo in eca
4 years ago
Ross Wightman ebf82b84ac
Merge pull request #122 from mrT23/master
4 years ago
Alexey Chernov bdb165a8a4 Merge changes in feature extraction interface to MobileNetV3
4 years ago
talrid 8a63c1add8 finalizing
4 years ago
talrid 6209146738 TResNet models
4 years ago
Ross Wightman 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
4 years ago
Ross Wightman 56608c9070 Bump version for pypi
4 years ago
Ross Wightman 1a8f5900ab Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
4 years ago
Chris Ha 06a50a94a8 Fix minor typos in create_attn.py and resnet.py
4 years ago
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 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
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 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 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