Skip to content

Instantly share code, notes, and snippets.

@rnyak

rnyak/NVT_dataset_object

Last active July 23, 2020 12:57
Show Gist options
  • Save rnyak/35a5c31e7b918d57c48445fe031c2bd5 to your computer and use it in GitHub Desktop.
Save rnyak/35a5c31e7b918d57c48445fe031c2bd5 to your computer and use it in GitHub Desktop.
Download ZIP
NVT_dataset_object
Raw
NVT_dataset_object
train_dataset = nvt.dataset(train_path, engine='parquet', gpu_memory_frac=0.3)
valid_dataset = nvt.dataset(valid_path, engine='parquet', gpu_memory_frac=0.3)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment