Add error logs to the training loop

[sweep-nightly]

Description

This PR adds error logs to the training loop in the main.py file. It ensures that any exceptions raised during training are logged with relevant information, allowing for easier debugging and error tracking.

Summary of Changes

• Added import statement for the logging module in main.py.
• Configured the logging module to write messages of level INFO and above to a file named training.log.
• Added try-except blocks in the training loop to catch and log any exceptions that may occur.
• Logged the exception messages along with any relevant information using the logging.error() function.
• Re-raised the exceptions after logging to ensure that the program does not silently continue after encountering an error.

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: Leftover TODOs in the code should be handled.
• Apply: All new business logic should have corresponding unit tests in the tests/ directory.
• Apply: Any clearly inefficient or repeated code should be optimized or refactored.