It looks like you have already some code written.
Where are you stuck? Do you get any PyTorch-related errors or is your model just not working?
I don’t think someone will completely implement a solution for you, especially as this seems to be your thesis.
PS: I’ve formatted your code and changed the topic for a clearer message.