Imported from https://github.com/lweitkamp/option-critic-pytorch