/EfficientNet.PyTorch

Concise, Modular, Human-friendly PyTorch implementation of EfficientNet with Pre-trained Weights.

Primary LanguagePythonMIT LicenseMIT

EfficientNet.PyTorch

This repository contains a concise, modular, human-friendly PyTorch implementation of EfficientNet with Pre-trained Weights.

Dependencies

Result Details(Val.)

Name # Params # FLOPS Top-1 Acc. Pretrained
efficientnet-b0 5.3M 0.39B 74.2 GoogleDrive
efficientnet-b1 7.8M 0.70B 78.0 GoogleDrive
efficientnet-b2 9.2M 1.0B 81.8 GoogleDrive
efficientnet-b3 12M 1.8B 82.7 GoogleDrive
efficientnet-b4 19M 4.2B 84.6 GoogleDrive
efficientnet-b5 30M 9.9B 86.1 GoogleDrive
efficientnet-b6 43M 19B 86.0 GoogleDrive
efficientnet-b7 66M 37B 86.7 GoogleDrive