6 lines
159 B
Python
6 lines
159 B
Python
from torch.utils.data import Dataset
|
|
|
|
|
|
class TemplateDataset(Dataset):
|
|
def __init__(self, *args, **kwargs):
|
|
super(TemplateDataset, self).__init__() |