Skip to content

Instantly share code, notes, and snippets.

@zaleslaw
Created June 9, 2022 13:13
Show Gist options
  • Save zaleslaw/18d35ea6de2afd65c481e1d320f01065 to your computer and use it in GitHub Desktop.
Save zaleslaw/18d35ea6de2afd65c481e1d320f01065 to your computer and use it in GitHub Desktop.
Loads data to the dataset
val trainDataset = OnHeapDataset.create(File(datasetPath, "train"), labelGenerator, preprocessing)
val valDataset = OnHeapDataset.create(File(datasetPath, "val"), labelGenerator, preprocessing)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment