merge dataset and sampler -> dataloader

pull/776/head
weishengyu 2021-06-03 15:14:05 +08:00
parent 96d1feb682
commit 65f0badf14
10 changed files with 5 additions and 6 deletions

View File

@ -18,14 +18,14 @@ from paddle.io import DistributedBatchSampler, BatchSampler, DataLoader
from ppcls.utils import logger
from . import dataset
from . import dataloader
from . import imaug
from . import samplers
# dataset
from .dataset.imagenet_dataset import ImageNetDataset
from .dataset.multilabel_dataset import MultiLabelDataset
from .dataset.common_dataset import create_operators
from .dataset.vehicle_dataset import CompCars, VeriWild
from .dataloader.imagenet_dataset import ImageNetDataset
from .dataloader.multilabel_dataset import MultiLabelDataset
from .dataloader.common_dataset import create_operators
from .dataloader.vehicle_dataset import CompCars, VeriWild
# sampler
from .samplers import DistributedRandomIdentitySampler

View File

@ -1 +0,0 @@
from .DistributedRandomIdentitySampler import DistributedRandomIdentitySampler