Refactor the model class: - [x] Add config - [ ] Separate Data loading - [ ] Separate training code
Refactor the model class: