Based on the notebook it seems you are working on an assignment and don’t provide much information here.
If you get stuck at a specific PyTorch-related problem, feel free to add some information to your post.