• Stars
    star
    153
  • Rank 243,368 (Top 5 %)
  • Language
    Dockerfile
  • Created over 8 years ago
  • Updated about 4 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Docker images for using torch-rnn

docker-torch-rnn

Docker images for using torch-rnn (https://github.com/jcjohnson/torch-rnn)

Available tags

  • crisbal/torch-rnn:base
    • Based on ubuntu:14.04
    • Allows usage of torch-rnn in CPU mode
  • lordalfred/docker-torch-rnn:cuda7.5
  • lordalfred/docker-torch-rnn:cuda8.0
  • lordalfred/docker-torch-rnn:cuda9.1
  • lordalfred/docker-torch-rnn:cuda9.2
  • lordalfred/docker-torch-rnn:cuda10.0 (requires nvidia-docker v2)
  • lordalfred/docker-torch-rnn:10.0-ubuntu18.04 (requires nvidia-docker v2)

How to

More details here: https://github.com/jcjohnson/torch-rnn#usage

CPU Only

  1. Start bash in the container

    • docker run --rm -ti crisbal/torch-rnn:base bash
  2. Preprocess the sample data

    python scripts/preprocess.py \
    --input_txt data/tiny-shakespeare.txt \
    --output_h5 data/tiny-shakespeare.h5 \
    --output_json data/tiny-shakespeare.json
    
  3. Train

    th train.lua \
    -input_h5 data/tiny-shakespeare.h5 \
    -input_json data/tiny-shakespeare.json \
    -gpu -1
    
  4. Sample

    • th sample.lua -checkpoint cv/checkpoint_10000.t7 -length 2000 -gpu -1

CUDA

  1. Install nvidia-docker

  2. Start bash in the container

    • nvidia-docker run --rm -ti lordalfred/docker-torch-rnn:cuda10.0 bash
  3. Preprocess the sample data

    python scripts/preprocess.py \
    --input_txt data/tiny-shakespeare.txt \
    --output_h5 data/tiny-shakespeare.h5 \
    --output_json data/tiny-shakespeare.json
    
  4. Train

    th train.lua \
    -input_h5 data/tiny-shakespeare.h5 \
    -input_json data/tiny-shakespeare.json
    
  5. Sample

    • th sample.lua -checkpoint cv/checkpoint_10000.t7 -length 2000

More Repositories

1

album-splitter

Split single-file MP3 albums into separate tracks. Downloads from YouTube supported.
Python
237
star
2

vue-webpack-ssr-fully-featured

A fully featured Vue.js boilerplate project that uses Vuex, vue-router, vue-i18n, Server Side Rendering and much, much more.
JavaScript
110
star
3

PlayStoreLinks_Bot

A Reddit Bot that links to Android Apps when requested.
Python
90
star
4

Appunti-Unimib

I miei appunti per il corso di Laurea in Informatica all'UniversitΓ  di Milano Bicocca
Jupyter Notebook
22
star
5

vue-keycloak

Use Keycloak Javascript Adapter in Vue.js
JavaScript
13
star
6

hmm-spellcheck

A spellchecker and corrector based on Hidden Markov Models
Python
12
star
7

Flask-Blog-Blueprint

A Flask extension/blueprint that adds a blog to your Flask webapp
Python
10
star
8

Titles-Youtube-Mp3

Search and download a list of videos from Youtube and convert them to mp3.
Python
10
star
9

kslf

Just a test
Shell
5
star
10

DotaCustomFilesJoiner

A tool for Dota2 Custom Files that makes your programming experience a little bit better.
Java
4
star
11

dotfiles

My dotfiles for an Arch Linux machine
Python
3
star
12

PHPNote

PHP service to write,save and share text notes.
PHP
3
star
13

dct-jpeg

Direct Cosine Transform function and JPEG-like image compression
Jupyter Notebook
3
star
14

Markov-Chain-Random-Text-Generator

A Random Text Generator based on the Markov Chain Algorithm
Python
3
star
15

redditClient

A reddit client written in python just for fun
2
star
16

TicToc.js

Simple JavaScript module to schedule events via a timer, with the ability to start and stop the timer.
JavaScript
2
star
17

Dota2TreasureSimulator

Python
2
star
18

crisbal.github.io

Personal Website and Blog
Vue
1
star
19

OnlineTicTacToe

Play TicTacToe Online in your browser, from every device. Made with NodeJs/Socket.io and HTML5/CSS3/JS
JavaScript
1
star
20

Twitter2Telegram_Bot

Send Tweets and get Twitter updates from Telegram.
JavaScript
1
star