Fix cannot import name MultiTaskDataSample

piercus · Nov 28, 2022 · 404d15d · 404d15d
1 parent 15fb9b0
commit 404d15d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)