File size: 294 Bytes
2b6a5b3 1392114 2b6a5b3 | 1 2 3 4 5 6 7 | # Description: Preload the dataset to cache_dir
# Copyright (C) 2024 Ronan Le Meillat
# License: Apache License 2.0
from datasets import load_dataset
dataset_id = "eltorio/ROCO-radiology"
cache_dir = "/workspace/data"
train_dataset = load_dataset(dataset_id, split="train", cache_dir=cache_dir) |