Commit Graph

17 Commits (27b3680d49e0e7c30ae6dd25f0daeb8de5319b40)

Author SHA1 Message Date
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 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 ba15ca47e8 Add ported EfficientNet-L2, B0-B7 NoisyStudent weights from TF TPU
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 5c4991a088 Add PyTorch trained EfficientNet-ES weights from Andrew Lavin
4 years ago
Ross Wightman 91534522f9 Add newly added TF ported EfficientNet-B8 weights (RandAugment)
4 years ago
Ross Wightman ec0dd4053a Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko
5 years ago
Ross Wightman acc3ed2b8c Add EfficientNet-B3 weights, trained from scratch with RA.
5 years ago
Ross Wightman 73b78459dc Add update RandAugment MixNet-XL weights
5 years ago
Ross Wightman ff421e5e09 New PyTorch trained EfficientNet-B2 weights with my RandAugment impl
5 years ago
Ross Wightman 3bef524f9c Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.
5 years ago
Ross Wightman 6ca0828166 Update EfficientNet comments, MobileNetV3 non-TF create fns, fix factory arg checks, bump PyTorch version req to 1.2
5 years ago
Ross Wightman eccbadca74 Update EfficientNet comments
5 years ago
Ross Wightman 902d32fb16 Renamed gen_efficientnet.py -> efficientnet.py
5 years ago