wehub-resource-sync
15 lines
651 B
Python
15 lines
651 B
Python
# coding: utf-8
|
|
|
|
from data.datasets_factory.dataset_x2v_interleave import X2VInterleaveIterableDataset
|
|
from data.datasets_factory.dataset_x2v_interleave_local import X2VInterleaveLocalIterableDataset
|
|
from data.datasets_factory.dataset_x2t_interleave import X2TInterleaveIterableDataset
|
|
from data.datasets_factory.dataset_x2t_interleave_local import X2TInterleaveLocalIterableDataset
|
|
|
|
DATASET_REGISTRY = {
|
|
# GEN
|
|
'x2v_interleave': X2VInterleaveIterableDataset,
|
|
'x2v_interleave_local': X2VInterleaveLocalIterableDataset,
|
|
'x2t_interleave': X2TInterleaveIterableDataset,
|
|
'x2t_interleave_local': X2TInterleaveLocalIterableDataset,
|
|
}
|