Red Chainer : A deep learning framework

A flexible framework for neural network for Ruby

Description

It ported python's Chainer with Ruby.

Requirements

• Ruby 2.4 or later

Installation

Add this line to your application's Gemfile:

gem 'red-chainer'

And then execute:

$ bundle

Or install it yourself as:

$ gem install red-chainer

Usage

Run MNIST example

MNIST sample program is here

# when install Gemfile
$ bundle exec ruby examples/mnist/mnist.rb
# when install yourself
$ ruby examples/mnist/mnist.rb

Run MNIST example with GPU

On GPU machine, add gem 'cumo' on Gemfile.local and do bundle install.

Run the example with --gpu option whose value indicates GPU device ID such as:

$ bundle exec ruby examples/mnist/mnist.rb --gpu 0

Development

Run tests

$ bundle exec ruby test/run_test.rb

Run tests with Cumo

On GPU machine, add gem 'cumo' on Gemfile.local and do bundle install.

Run tests with RED_CHAINER_GPU environment variable whose value indicates GPU device ID such as:

$ bundle exec env RED_CHAINER_GPU=0 ruby test/run_test.rb

License

The MIT license. See LICENSE.txt for details.

Red Chainer implementation status