diff --git a/mmcls/structures/__init__.py b/mmcls/structures/__init__.py index 6d6a8af3f99..3021d0a7d0b 100644 --- a/mmcls/structures/__init__.py +++ b/mmcls/structures/__init__.py @@ -1,5 +1,6 @@ # Copyright (c) OpenMMLab. All rights reserved. from .cls_data_sample import ClsDataSample +from .multi_task_data_sample import MultiTaskDataSample from .utils import (batch_label_to_onehot, cat_batch_labels, stack_batch_scores, tensor_split)