Add new CNN class for MNIST handling and import it into main.py

[sweep-nightly]

Description

This PR adds a new CNN class in a new file named cnn.py to handle the MNIST dataset. The CNN class is implemented using PyTorch and inherits from torch.nn.Module. It includes the necessary layers and methods for training the model on the MNIST dataset.

The existing main.py file is modified to import the new CNN class and replace the current model definition with an instance of the CNN class. The training code is also updated to use the training method of the CNN class.

Summary of Changes

• Created a new file src/cnn.py to define the CNN class for handling the MNIST dataset.
• Imported the necessary modules in cnn.py and defined the CNN class with the required layers and methods.
• Modified main.py to import the new CNN class from cnn.py and replace the current model definition with an instance of the CNN class.
• Updated the training code in main.py to use the training method of the CNN class.

Please review and merge this PR to incorporate the new CNN class for improved handling of the MNIST dataset.

Fixes #9.

---

🎉 Latest improvements to Sweep:

• Sweep can now passively improve your repository! Check out Rules to learn more.

---

💡 To get Sweep to edit this pull request, you can:

• Comment below, and Sweep can edit the entire PR
• Comment on a file, Sweep will only modify the commented file
• Edit the original issue to get Sweep to recreate the PR from scratch

[sweep-nightly]

Rollback Files For Sweep

• Rollback changes to src/main.py
• Rollback changes to src/cnn.py

[sweep-nightly]

Apply Sweep Rules to your PR?

• Apply: All docstrings and comments should be up to date.
• Apply: Code should be properly formatted and indented.
• Apply: Variable and function names should be descriptive and follow a consistent naming convention.
• Apply: Imports should be organized and grouped together.
• Apply: There should be no unused imports or variables.
• Apply: Code should be properly commented and include docstrings for functions and classes.