Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 

rnn_zoo

This repository tests various recurrent neural network architectures on baseline datasets SeqMNIST and pMNIST.
The network architectures chosen were those deemed to be the most effective currently available.

Architectures tested include:

  • RNN

  

    

    

  

    

    

Results

The following results are were generated using the architectures listed above.
Hyperparameters used: layers 3, num neurons 50, optimizer Adam, learning rate .0001 and batch 64.




Running the code

  python train.py --model-type=irnn --task=seqmnist --layers=2 --batch-size=64 --epochs=10

Installing

Update BASE_DIR in config.ini with the absolute path to the current directory.
Packages needed to run the code include:

  • numpy
  • python
  • PyToch
  • argparse
  • configparser

About

This repository tests various recurrent neural network architectures on baseline datasets SeqMNIST and pMNIST.

Resources

Releases

No releases published

Packages

No packages published

Languages