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)}.")