Could you please format the post by wrapping the code and error messages into three backticks ``` as in the previous posts?
It’s currently hard to debug and I can’t find any GPU usage, so I’m still unsure why you would like to push the parameters to the GPU at all.