Imported from https://github.com/zubair-irshad/Imitation-learning