← all repositories

rwightman/gen-efficientnet-pytorch

PyTorch implementations of EfficientNet, MobileNetV3, MixNet and other efficient CNN architectures with pretrained weights for image classification.

1.6k stars Python Computer VisionML Frameworks
gen-efficientnet-pytorch
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This repository provides a collection of efficient convolutional neural network architectures for image classification, including EfficientNet variants, EfficientNet-Lite, MobileNetV2/V3, MixNet, MNASNet, FBNet, and Single-Path NAS. All models are implemented in PyTorch with pretrained weights and support for ONNX and Caffe2 export. The library has been deprecated in favor of the more comprehensive timm library which supersedes it.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.