Cleaned some code
Closes #14 (closed)
- removed train-test-split in the Dataset classes
- jitted the train step func directly and removed decorator
Fraunhofer GitLab CE, EE, Harbor Registry and Mattermost will be unavailable on Wednesday, December 17th, between 4 PM and 8 PM CET - at the latest. Please log out during this time to avoid potential data loss.
Closes #14 (closed)