Requirements - python and pytorch

[jsrozner]

Hi, I think this approach will work for our research problem and would like to adapt this repository. Is this still a good place to start?

Assuming this is a good starting point, do you remember what python and pytorch versions you were using? I assume python 2 because there's a print x statement. But I'm running into some bugs when I try to run the base version (with --debug) because of pytorch changes.

Thanks!

[jsrozner]

I had to make the following small changes to get the code to run:
jsrozner@70e59d4

I can pull request if you'd like? But please make sure they don't seem to break anything. Notes in the commit message about the changes I had to make, but I think it's mostly an issue with pytorch's handling of zero-dimensional tensors.