From cc5c336183b3bea75f7c4773c5bd0210d1f79b8a Mon Sep 17 00:00:00 2001 From: liuht Date: Wed, 15 Nov 2023 17:03:38 +0800 Subject: [PATCH] [FIX] delete print --- learnware/market/heterogeneous/organizer/hetero_map/__init__.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/learnware/market/heterogeneous/organizer/hetero_map/__init__.py b/learnware/market/heterogeneous/organizer/hetero_map/__init__.py index dc3b90e..37b5d3e 100644 --- a/learnware/market/heterogeneous/organizer/hetero_map/__init__.py +++ b/learnware/market/heterogeneous/organizer/hetero_map/__init__.py @@ -274,9 +274,7 @@ class HeteroMap(nn.Module): if isinstance(x, pd.DataFrame): inputs = self.feature_tokenizer(x) elif isinstance(x, torch.Tensor): - logger.info(f"extract features, input device:{x.device}") inputs = self.feature_tokenizer.forward(cols, x) - logger.info(f"extract features, output device:{inputs['x_num'].device}") else: raise ValueError(f"feature_tokenizer takes inputs with dict or pd.DataFrame, find {type(x)}.")