Hi techboy,
What you should do to fix this error is to convert the tensor which is in dtype Long to float. You can figure out which tensor is causing error by looking out the line which gives error while execution.