STAckable-Recurrent-network icon indicating copy to clipboard operation
STAckable-Recurrent-network copied to clipboard

[PAMI 2021] Gating Revisited: Deep Multi-layer RNNs That Can Be Trained

trafficstars

STAckable Recurrent (STAR) Network

[PAMI21] Tensorflow implementation for STAckable Recurrent (STAR) network based on our TPAMI paper:

Gating Revisited: Deep Multi-layer RNNs That Can Be Trained.

[Paper] - [Blog post]

Getting Started

Run the model with

python main.py --cell bn-star --stack 4 --data pmnist

Citation

@article{turkoglu2021gating,
  title={Gating revisited: Deep multi-layer rnns that can be trained},
  author={Turkoglu, Mehmet Ozgur and D'Aronco, Stefano and Wegner, Jan and Schindler, Konrad},
  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
  year={2021},
  publisher={IEEE}
}

The code is mostly based on https://github.com/JosvanderWesthuizen/janet.