https://github.com/hvy/chainer-gan-trainer

Chainer implementation of the Generative Adversarial Network using the training module
https://github.com/hvy/chainer-gan-trainer

Last synced: 8 days ago
JSON representation

Chainer implementation of the Generative Adversarial Network using the training module

• Host: GitHub
• URL: https://github.com/hvy/chainer-gan-trainer
• Owner: hvy
• Created: 2016-10-25T09:06:11.000Z (about 8 years ago)
• Default Branch: master
• Last Pushed: 2016-11-09T08:45:21.000Z (about 8 years ago)
• Last Synced: 2024-10-30T06:59:13.649Z (about 2 months ago)
• Language: Python
• Homepage:
• Size: 11.7 KB
• Stars: 3
• Watchers: 2
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Chainer GAN Trainer Demo

Implementation of GAN with Chainer's *trainer* using the following custom classes.

- `chainer.iterators`

- `chainer.training.updater`

- `chainer.training.extensions`

## Custom Trainer Components

### chainer.iterators.RandomNoiseIterator

Iterator that keeps producing random arrays of Gaussian or uniform distribution. It has no notion of epochs.

### chainer.training.updater.GenerativeAdversarialUpdater

Updater responsible for the GAN training algorithm including forward pass, backward pass and parameter updates.

### chainer.training.extensions.GeneratorSample

Extension that automatically takes random sample images and saved them to disk at any given interval.

## Run

### Train

```bash

python train.py --gpu 0

```