Skip to content

Commit c68dc92

Browse files
committed
add dropout with if
1 parent 1665d68 commit c68dc92

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

rasa/nlu/classifiers/diet_classifier.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1193,9 +1193,12 @@ def _combine_sparse_dense_features(
11931193
_f = f
11941194

11951195
dense_f = self._tf_layers[f"sparse_to_dense.{name}"](_f)
1196-
dense_f = self._tf_layers[f"dense_input_dropout.{name}"](
1197-
dense_f, self._training
1198-
)
1196+
1197+
if dense_dropout:
1198+
dense_f = self._tf_layers[f"dense_input_dropout.{name}"](
1199+
dense_f, self._training
1200+
)
1201+
11991202
dense_features.append(dense_f)
12001203
else:
12011204
dense_features.append(f)

0 commit comments

Comments
 (0)