File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
examples/text_classification/rnn Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -25,7 +25,7 @@ class BoWModel(nn.Layer):
25
25
"""
26
26
This class implements the Bag of Words Classification Network model to classify texts.
27
27
At a high level, the model starts by embedding the tokens and running them through
28
- a word embedding. Then, we encode these epresentations with a `BoWEncoder`.
28
+ a word embedding. Then, we encode these representations with a `BoWEncoder`.
29
29
Lastly, we take the output of the encoder to create a final representation,
30
30
which is passed through some feed-forward layers to output a logits (`output_layer`).
31
31
Original file line number Diff line number Diff line change @@ -117,7 +117,7 @@ def create_dataloader(dataset,
117
117
padding_idx = pad_token_id )
118
118
elif network == 'bilstm_attn' :
119
119
lstm_hidden_size = 196
120
- attention = SelfInteractiveAttention (hidden_size = 2 * stm_hidden_size )
120
+ attention = SelfInteractiveAttention (hidden_size = 2 * lstm_hidden_size )
121
121
model = BiLSTMAttentionModel (
122
122
attention_layer = attention ,
123
123
vocab_size = vocab_size ,
You can’t perform that action at this time.
0 commit comments