PVD/dataset/test_rotor37_data.py
2023-04-11 16:00:54 +02:00

12 lines
291 B
Python

import datasets
train_ds = datasets.load_dataset("dataset/rotor37_data.py", split="train")
train_ds = train_ds.with_format("torch")
print(train_ds)
test_ds = datasets.load_dataset("dataset/rotor37_data.py", split="test")
test_ds = test_ds.with_format("torch")
print(test_ds)
print("yay")