Skip to content

Commit ab89f09

Browse files
committed
remove print statement
1 parent f6e2341 commit ab89f09

1 file changed

Lines changed: 0 additions & 2 deletions

File tree

mambular/base_models/tabtransformer.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,6 @@ def forward(self, num_features, cat_features):
199199
cat_embeddings = self.embedding_norm(cat_embeddings)
200200
else:
201201
cat_embeddings = None
202-
print(cat_embeddings.shape)
203202

204203
num_features = torch.cat(num_features, dim=1)
205204
num_embeddings = self.norm_f(num_features)
@@ -215,7 +214,6 @@ def forward(self, num_features, cat_features):
215214
else:
216215
raise ValueError(f"Invalid pooling method: {self.pooling_method}")
217216

218-
print(num_embeddings.shape, x.shape)
219217
x = torch.cat((x, num_embeddings), axis=1)
220218
preds = self.tabular_head(x)
221219

0 commit comments

Comments
 (0)