Skip to content

Cleaned some code

Closes #14 (closed)

  • removed train-test-split in the Dataset classes
  • jitted the train step func directly and removed decorator

Merge request reports