There is not nearly enough information in your question to start thinking about how to help you!
Please remember that we don’t know anything about the problem that you are facing. Please provide enough context so that someone who has no clue about what you are dealing with (but has some experience working with ML and PyTorch) can understand the issue.