Skip to content
This repository has been archived by the owner on Jan 3, 2023. It is now read-only.

NervanaSystems/ModelZoo

Repository files navigation

DISCONTINUATION OF PROJECT

This project will no longer be maintained by Intel. Intel has ceased development and contributions including, but not limited to, maintenance, bug fixes, new releases, or updates, to this project. Intel no longer accepts patches to this project. This repo contains a number of standard deep learning models that can be run with the neon libraries. The specific model type subdirectories include more detailed information on the various models. Below is a summary of the various tasks and models included here that address those tasks.

Image classification

ImageNet

  • Googlenet
  • VGG
  • Alexnet

CIFAR-10

  • Deep Residual Network
  • All CNN

Object localization

  • Fast-RCNN

Scene Classification

  • Deep Residual Network

Image Captioning

  • LSTM

Video Games

  • Deep Q-network

NLP

bAbI (Question Answering)

  • GRU/LSTM model

Sentiment classification

  • LSTM (IMDB dataset)

Video

  • C3D (UCF101 dataset)

Transfer Learning

  • Classification using multi scale sampling (PASCALVOC dataset)

Speech

  • Deep Speech 2 (Librispeech dataset)