Skip to content

Commit 929198c

Browse files
committed
All tests passed. For the time being mutiple tabular components will not be offered to the 'load from folder' classes. Therefore, only thing left is adjust docs and test in GPU
1 parent 3d25efd commit 929198c

File tree

2 files changed

+422
-564
lines changed

2 files changed

+422
-564
lines changed

pytorch_widedeep/models/wide_deep.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ def _forward_deephead(
345345
if self.is_tabnet:
346346
deepside, M_loss = self.deeptabular(X["deeptabular"])
347347
else:
348-
deepside = self._forward_component(
348+
deepside = self._forward_component_with_head(
349349
X, self.deeptabular, "deeptabular", deepside
350350
)
351351

0 commit comments

Comments
 (0)