Skip to content
Please do not feed the models
Branch: master
Clone or download
Latest commit fabb071 Mar 21, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
audio/speech-blstm Add variable i to count which epoch program is in Dec 7, 2018
games/differentiable-programming update gym toml- pendulum Mar 13, 2019
other flux vnext intro Mar 12, 2019
script tutorials folder Oct 24, 2018
text
tutorials
vision Merge pull request #89 from avik-pal/ap/cifar_fixes Feb 8, 2019
.gitignore
LICENSE.md move things around Jul 12, 2018
README.md

README.md

Flux Model Zoo

This repository contains various demonstrations of the Flux machine learning library. Any of these may freely be used as a starting point for your own models.

The models are broadly categorised into the folders vision (e.g. large convolutional neural networks (CNNs)), text (e.g. various recurrent neural networks (RNNs) and natural language processing (NLP) models), games (Reinforcement Learning / RL). See the READMEs of respective models for more information.

Usage

Each folder is its own Julia project, which lists the packages you need to run the models. You can run the models by opening Julia in the project folder and running

using Pkg; Pkg.activate("."); Pkg.instantiate()

to install all needed packages. Then you can run the model code with include("script.jl") or by running the script line-by-line. More details are available in the README for each model.

Models with a cuda folder can be loaded with NVIDIA GPU support, if you have a CUDA installed.

using Pkg; Pkg.activate("cuda"); Pkg.instantiate()
using CuArrays

Contributing

We welcome contributions of new models. They should be in a folder with a project and manifest file, to pin all relevant packages, as well as a README to explain what the model is about, how to run it, and what results it achieves (if applicable). If possible models should not depend directly on GPU functionality, but ideally should be CPU/GPU agnostic.

Model Listing

You can’t perform that action at this time.