Skip to content

Commit

Permalink
Merge branch 'master' into tensorflow-addons-version
Browse files Browse the repository at this point in the history
  • Loading branch information
alwx authored Mar 3, 2020
2 parents 3fa1d40 + 791a899 commit db0830f
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 5 deletions.
2 changes: 1 addition & 1 deletion docs/nlu/about.rst
Original file line number Diff line number Diff line change
Expand Up @@ -26,4 +26,4 @@ and returning structured data like
}
}
Rasa NLU used to be a separate library, but it is now part of the Rasa framework.
If you want to use Rasa NLU on its own, see :ref:`using-nlu-only`.
10 changes: 6 additions & 4 deletions rasa/nlu/classifiers/diet_classifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -1214,16 +1214,18 @@ def _create_sequence(
features, mask, name, sparse_dropout=self.config[SPARSE_INPUT_DROPOUT]
)

x = self._tf_layers[f"ffnn.{name}"](inputs, self._training)
inputs = self._tf_layers[f"ffnn.{name}"](inputs, self._training)

if masked_lm_loss:
x, lm_mask_bool = self._tf_layers[f"{name}_input_mask"](
x, mask, self._training
inputs, lm_mask_bool = self._tf_layers[f"{name}_input_mask"](
inputs, mask, self._training
)
else:
lm_mask_bool = None

outputs = self._tf_layers[f"{name}_transformer"](x, 1 - mask, self._training)
outputs = self._tf_layers[f"{name}_transformer"](
inputs, 1 - mask, self._training
)
outputs = tfa.activations.gelu(outputs)

return outputs, inputs, seq_ids, lm_mask_bool
Expand Down

0 comments on commit db0830f

Please sign in to comment.