Failure when running test_genre.sh

[ErinKoch2]

Hello, I am attempting to run test_genre.sh. At first it was not clear that I needed to provide a gpu argument ( I think this is supposed to be an index???). When I added this it seems to get through most of the set-up, but then fails during the actually testing. I providing a screen shot what I ran in the terminal and of the error traceback. Do you have any insight into this error? I have been trying for quite some time to resolve it with no success. Ultimately I have my own data set that I would like to run through the trained the model. Please let me know. Thanks!

[xiumingzhang]

Does this help? https://github.com/pytorch/pytorch/issues/7548