farhan0syakir / vanilla-memory-network-pytorch

This is my implementation of minimum memory network using pytorch

vanilla memory network pytorch

This is my implementation of minimum permanent memory network using pytorch

ideas from paper: Mantra: memory augmented networks for multiple trajectory prediction

Running

python train.py