Add error logs to the training loop

[sweep-nightly]

Description

This PR adds error logs to the training loop in the src/main.py file. The training loop is responsible for training the model over a certain number of epochs. With the addition of error logs, any exceptions or errors that occur during the training process will be logged, providing useful information for debugging and tracking the progress of the training.

Summary of Changes

• Imported the logging module at the beginning of the src/main.py file.
• Configured the logging module to output logs to a file named training.log with a level of INFO.
• Added a try-except block around the code that could potentially raise an exception in the training loop.
• In the except block, logged the exception using the logging module.
• Also logged the current epoch and batch number in the except block for context.

Please review and merge this PR. Thank you!

Fixes #1.

---

🎉 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

[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.