File size: 1,125 Bytes
c3396ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
---
license: mit
datasets:
- ILSVRC/imagenet-1k
pipeline_tag: image-classification
---
# Introduction
This repository stores the model for Efficientnet-b4, compatible with Kalray's neural network API. </br>
Please see www.github.com/kalray/kann-models-zoo for details and proper usage. </br>
Please see https://huggingface.co/docs/transformers/main/en/model_doc/efficientnet for Efficientnet model description. </br>
# Contents
- ONNX: efficientNet-b4.onnx
# Lecture note reference
- EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks, https://arxiv.org/pdf/1905.11946
# Repository or links references
- https://pytorch.org/vision/stable/models/generated/torchvision.models.efficientnet_b4.html#torchvision.models.efficientnet_b4
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}
}
```
Author: nbouberbachene@kalrayinc.com |