Commit Graph

374 Commits (ebf82b84ac13d3aeed1714a67a5c52cf1193bbad)
 

Author SHA1 Message Date
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
Chris Ha 89a32454e1 Merge remote-tracking branch 'upstream/master'
5 years ago
Ross Wightman 7d07ebb660 Adding some configs to sknet, incl ResNet50 variants from 'Compounding ... Assembled Techniques' paper and original SKNet50
5 years ago
Ross Wightman a9d2424fd1 Add separate zero_init_last_bn function to support more block variety without a mess
5 years ago
Ross Wightman 355aa152d5 Just leave it float for now, will look at fp16 later. Remove unused reference code.
5 years ago
Ross Wightman ef457555d3 BlockDrop working on GPU
5 years ago
Ross Wightman 3ff19079f9 Missed nn_ops.py from last commit
5 years ago
Ross Wightman 9f11b4e8a2 Add ConvBnAct layer to parallel integrated SelectKernelConv, add support for DropPath and DropBlock to ResNet base and SK blocks
5 years ago
Ross Wightman cefc9b7761 Move SelectKernelConv to conv2d_layers and more
5 years ago
Ross Wightman 9abe610931 Used wrong channel var for split
5 years ago
Ross Wightman 58e28dc7e7 Move Selective Kernel blocks/convs to their own sknet.py file
5 years ago
Ross Wightman a93bae6dc5 A SelectiveKernelBasicBlock for more experiments
5 years ago
Ross Wightman ad087b4b17 Missed bias=False in selection conv
5 years ago
Ross Wightman c8b3d6b81a Initial impl of Selective Kernel Networks. Very much a WIP.
5 years ago
Ross Wightman fd98fb33c5 Update sotabench with tf_efficientnet_b8 model
5 years ago
Ross Wightman 7c88356682 Add update results on ImageNet validation, ImageNetV2, ImageNet-A, and ImageNet-Sketch for all models
5 years ago
Ross Wightman 1daa303744 Add support to Dataset for class id mapping file, clean up a bit of old logic. Add results file arg for validation and update script.
5 years ago
Ross Wightman 91534522f9 Add newly added TF ported EfficientNet-B8 weights (RandAugment)
5 years ago
Ross Wightman 82dd60b33c Rename results files for more clarity
5 years ago
Ross Wightman 5b7cc16ac9 Add warning about using sync-bn with zero initialized BN layers. Fixes #54
5 years ago
Ross Wightman b18c19901e
Update README.md
5 years ago
Ross Wightman 12dbc74742 New ResNet50 JSD + RandAugment weights
5 years ago
Chris Ha 0697ab183b Update .gitignore
5 years ago
Ross Wightman e39aae56b4
Update README.md
5 years ago
Ross Wightman 7a17ee9c3b Update README.md
5 years ago