Welcome to Pytorch forum!
It would be great if you could give a bit more detailed information with your question and the exact error message that you get. Without that, there is less chance that you will receive any valuable direction to solve your issue. Also, a meaningful title doesn’t hurt as well. Happy coding!