Browse Source

[FIX] delete print

tags/v0.3.2
liuht 2 years ago
parent
commit
cc5c336183
1 changed files with 0 additions and 2 deletions
  1. +0
    -2
      learnware/market/heterogeneous/organizer/hetero_map/__init__.py

+ 0
- 2
learnware/market/heterogeneous/organizer/hetero_map/__init__.py View File

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



Loading…
Cancel
Save