Imported from https://github.com/garroud/human-motion-prediction-pytorch