Update DataLoader Usage Example in Documentation (Fixes #110)

[agentmarketbot]

Pull Request Description

Overview

This pull request addresses the feedback provided in Issue #110 concerning the DataLoader usage example in the repository's documentation. The original example did not properly utilize the enumerate function, which is essential for tracking the index of each batch while iterating through the data loader.

Changes Made

• Updated DataLoader Example: The example in the documentation was modified to include the enumerate() function. This change ensures that users can correctly obtain both the batch index and the batch contents during iteration.

Code Changes

The documentation in the README.md file has been updated from:

for batch_idx, (x_batch, y_batch) in data_loader:
    pass

To:

for batch_idx, (x_batch, y_batch) in enumerate(data_loader):
    pass

Additional Information

• This change was prompted by user feedback that highlighted the need for clarity in the DataLoader example. Adding enumerate enhances the usability of the provided code snippet for developers who may not be familiar with how to manage batch indices effectively.
• The change has been reviewed for consistency with the overall documentation style and correctness.

Conclusion

With these updates, the DataLoader example is now more intuitive and functional for users referencing the documentation.

Fixes #110.

Please review the changes and let me know if there are any further modifications required or if any additional documentation needs to be updated!