Bug Report for handwritten-digit-classifier

[sara-02]

Bug Report for https://neetcode.io/problems/handwritten-digit-classifier

Please describe the bug below and include any steps to reproduce the bug or screenshots if possible.

Even when using the exact code given in the solutions, we see that the test case is failing

images=torch.randn(1, 28 * 28)
[[0.607,0.4426,0.494,0.4634,0.5263,0.5707,0.5564,0.4835,0.3663,0.5077]]
[[0.565,0.4828,0.5203,0.471,0.5244,0.5394,0.5588,0.4864,0.4046,0.5196]]