Commit b1ad92b8 authored by Orion Reblitz-Richardson's avatar Orion Reblitz-Richardson
Browse files

Update neural_gpu documentation.

* Add list of algorithmic tasks.
* Detail what is stored in /tmp/neural_gpu/ (default location).
* Detail that one should hit Ctrl-C to stop the training process.
* Add sections around decoding and animation, but couldn't get those to work myself. Still worth having in the docs.
parent f8cc1ea5
...@@ -6,6 +6,50 @@ Requirements: ...@@ -6,6 +6,50 @@ Requirements:
* TensorFlow (see tensorflow.org for how to install) * TensorFlow (see tensorflow.org for how to install)
* Matplotlib for Python (sudo apt-get install python-matplotlib) * Matplotlib for Python (sudo apt-get install python-matplotlib)
Run: python neural_gpu_trainer.py --task=rev The model can be trained on the following algorithmic tasks:
* `sort` - Sort a decimal list
* `kvsort` - Sort decimal keys in dictionary
* `id` - Return the same decimal list
* `rev` - Reverse a decimal list
* `rev2` - Reverse a decimal dictionary by key
* `incr` - Add one to a decimal
* `add` - Long decimal addition
* `left` - First decimal in list
* `right` - Last decimal in list
* `left-shift` - Left shift a decimal list
* `right-shift` - Right shift a decimal list
* `bmul` - Long binary multiplication
* `mul` - Long decimal multiplication
* `dup` - Duplicate a decimal list with padding
* `badd` - Long binary addition
* `qadd` - Long quaternary addition
* `search` - Search for decimal key in dictionary
To train the model on the reverse task run:
```
python neural_gpu_trainer.py --task=rev
```
While training, interim / checkpoint model parameters will be
written to `/tmp/neural_gpu/`.
Once the amount of error gets down to what you're comfortable
with, hit `Ctrl-C` to stop the training process. The latest
model parameters will be in `/tmp/neural_gpu/neural_gpu.ckpt-<step>`
and used on any subsequent run.
To test a trained model on how well it decodes run:
```
python neural_gpu_trainer.py --task=rev --mode=1
```
To produce an animation of the result run:
```
python neural_gpu_trainer.py --task=rev --mode=1 --animate=True
```
Maintained by Lukasz Kaiser (lukaszkaiser) Maintained by Lukasz Kaiser (lukaszkaiser)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment