Arnold
Arnold copied to clipboard
Not support PyTorch 0.4.0
I test the code using PyTorch 0.4.0 under windows, it occurs an error beacase of the changed 'Variable' in PyTorch 0.4.0. Then I test on PyTorch 0.3.1, it's ok.
Fixed the assertions to make it work with Python3 and PyTorch 1.3.1. Here the PR: https://github.com/glample/Arnold/pull/10