13 lines
251 B
Python
13 lines
251 B
Python
from torch.utils.data import Dataset
|
|
|
|
|
|
class TemplateDataset(Dataset):
|
|
def __init__(self, *args, **kwargs):
|
|
super(TemplateDataset, self).__init__()
|
|
|
|
def __len__(self):
|
|
pass
|
|
|
|
def __getitem__(self, item):
|
|
return item
|