camel
copied
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
11 lines
351 B
11 lines
351 B
3 years ago
|
from .field import RawField, Merge, ImageField, TextField
|
||
|
|
from .dataset import *
|
||
|
|
from torch.utils.data import DataLoader as TorchDataLoader
|
||
|
|
|
||
|
|
from .dataset import *
|
||
|
|
|
||
|
|
|
||
|
|
class DataLoader(TorchDataLoader):
|
||
|
|
def __init__(self, dataset, *args, **kwargs):
|
||
|
|
super(DataLoader, self).__init__(dataset, *args, collate_fn=dataset.collate_fn(), **kwargs)
|