What version of PyTorch are you using? I am having the following problems with it.  Can you help with this? Thank you very much!