Rewriting train config Configuration
Summary
This MR introduces the following changes
- Adding dataclasses for the train_config.
Relates to the following issues
- Closes #208 (closed)
- #181 (closed) for the train configs
This MR introduces the following changes
Relates to the following issues