From 2ff7e15248b513702379a870fb389bf006806c8a Mon Sep 17 00:00:00 2001 From: liuht Date: Wed, 6 Dec 2023 15:51:20 +0800 Subject: [PATCH] [FIX] fix wrong delete --- learnware/market/heterogeneous/organizer/hetero_map/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/learnware/market/heterogeneous/organizer/hetero_map/__init__.py b/learnware/market/heterogeneous/organizer/hetero_map/__init__.py index 4ebb0a8..e124bcb 100644 --- a/learnware/market/heterogeneous/organizer/hetero_map/__init__.py +++ b/learnware/market/heterogeneous/organizer/hetero_map/__init__.py @@ -9,7 +9,7 @@ from torch import Tensor, nn from .....utils import allocate_cuda_idx, choose_device from .....specification import HeteroMapTableSpecification, RKMETableSpecification from .feature_extractor import CLSToken, FeatureProcessor, FeatureTokenizer -from .trainer import TransTabCollatorForCL +from .trainer import TransTabCollatorForCL, Trainer class HeteroMap(nn.Module):