Performance Optimizations, Code Cleanup, and Enhanced Usability

[jiteshm17]

This pull request introduces several improvements and optimizations to enhance the performance, maintainability, and usability of the text segmentation repository.

Summary of Changes:

• Updated code for Python 3 compatibility and latest PyTorch versions (15c6687).
• Moved parts of the model logic to collate_fn to streamline code execution & significant speed improvements** by removing loops in the forward method of both the base LSTM and segmentation LSTM models. This optimization provides orders of magnitude faster execution when working with larger batch sizes (c6fffb01, 0ef4d15 and badbfb0).
• Minor bug fixes and state saving improvements (9039a3a) (Oct 12, 2024).
• Exception handling while training and testing to skip faulty batches to enhance robustness (f2fbf76).