Imported from https://github.com/navneet-nmk/pytorch-rl