Kalray/efficientnet-b4

Introduction

This repository stores the model for Efficientnet-b4, compatible with Kalray's neural network API.
Please see www.github.com/kalray/kann-models-zoo for details and proper usage.
Please see https://huggingface.co/docs/transformers/main/en/model_doc/efficientnet for Efficientnet model description.

Contents

• ONNX: efficientNet-b4.onnx
• Quantized ONNX (INT8): efficientNet-b4-q.onnx

Lecture note reference

• EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks, https://arxiv.org/pdf/1905.11946

Repository or links references

• PyTorch | TorchVision

BibTeX entry and citation info

@inproceedings{he2016deep,
  title     = {Deep residual learning for image recognition},
  author    = {He, Kaiming and Zhang, Xiangyu and Ren, Shaoqing and Sun, Jian},
  booktitle = {Proceedings of the IEEE conference on computer vision and pattern recognition},
  pages     = {770--778},
  year      = {2016}
}

Authors:

• qmuller@kalrayinc.com
• nbouberbachene@kalrayinc.com