Problem with your code

[marcomameli1992]

Dear, I'm trying to use your code but there are some problems with the pytorch version. You use the version 0.4 but this version is very old and in my computer I can't use it due to the CUDA version that I need for other projects. Can you provide some updates withouth the use of nn.legacy that is deprecated from pytorch 1.0

Thank you.