Skip to content
The Statistical Recurrent Unit in Pytorch
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

The Statistical Recurrent Unit

  • authors: Junier B. Oliva, Barnabas Poczos, Jeff Schneider
  • arxiv:
  • Pytorch implemention of the experiment of SRU with pixel-by-pixel sequential MNIST.
  • Powered by DL HACKS


  • environment: python3.5
  • pytorch 0.2.0
  • hyperopt 0.1
  • numpy 1.13.1
  • scikit-learn 0.18.2


  • python sru: trainning RNNs with fixed parameters.
  • python sru : tuning hyper parameters with hyperopt.
  • Choose your model from [sru, gru, lstm]
  • If you need more information, please run python --help.


  • I choose Adam for optimization, though SGD is used in the paper. (It might converge faster)
  • weight_decay is used. (The paper doesn't refer to it)
You can’t perform that action at this time.