HaoMood / pytorch-utilities

Data loader and other utility files for PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch-utilities

Data loader and other utility files for PyTorch.

Datasets

All datasets are subclasses of torch.utils.data.Dataset i.e, they have _getitem_ and _len_ methods implemented. Hence, they can all be passed to a torch.utils.data.DataLoader which can load multiple samples parallelly using torch.multiprocessing workers. All the datasets have almost similar API. They all have a common arguments: transform to transform the input.

About

Data loader and other utility files for PyTorch


Languages

Language:Python 100.0%