Optimising Model Parameters Using PyTorch