Skip to content

Burton2000/pytorch_rnn_sequence_generation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RNN time series generation experiments

This repo provides an example of training an RNN/GRU model to overfit and generate a sin wave that it's trained on. Once trained we supply an initial sequence of points and the model will then continue to generate the sin wave its learnt.

Obviously this isn't a particularly difficult task but is meant as just a way to demonstrate simple use of RNNs in PyTorch.

How to use

  • Run train.py file, your model will be trained and then will automatically generate points after training.

This is the result after training.

Alt text

Updated for PyTorch 0.4.1

About

Train an RNN using PyTorch to generate a sin wave

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages