-
-
Notifications
You must be signed in to change notification settings - Fork 385
/
__init__.py
41 lines (40 loc) · 1006 Bytes
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# flake8: noqa
from catalyst.data.augmentor import Augmentor, AugmentorCompose, AugmentorKeys
from catalyst.data.collate_fn import FilteringCollateFn
from catalyst.data.dataset import (
DatasetFromSampler,
ListDataset,
MergeDataset,
NumpyDataset,
PathsDataset,
MetricLearningTrainDataset,
QueryGalleryDataset,
)
from catalyst.data.loader import (
ILoaderWrapper,
BatchLimitLoaderWrapper,
BatchPrefetchLoaderWrapper,
)
from catalyst.data.reader import (
ReaderSpec,
ScalarReader,
LambdaReader,
ReaderCompose,
)
from catalyst.data.sampler import (
BalanceClassSampler,
BalanceBatchSampler,
DistributedSamplerWrapper,
DynamicLenBatchSampler,
DynamicBalanceClassSampler,
MiniEpochSampler,
)
from catalyst.data.sampler_inbatch import (
IInbatchTripletSampler,
InBatchTripletsSampler,
AllTripletsSampler,
HardTripletsSampler,
HardClusterSampler,
)
from catalyst.data.cv import *
from catalyst.data.nlp import *