Unable to run the training code

Hi,
Thanks for a very interesting work, I want to train the code on my own dataset but I can't find a suitable torch version to support the code without changes.
Can you share the versions you used to train the code?
Thank you,
Nimrod