Can I know what are the two arguments that throw up the arguments located on different gpu error? Printing the stack trace and all the lines that deal with pushing the variables and model to the gpu, will also immensely help.