Skip to content

Commit e192179

Browse files
committed
Fix typo (#42)
Signed-off-by: fred@pegasus <[email protected]>
1 parent 52b1bb4 commit e192179

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pocket/models/transformers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def forward(self,
206206
x = self.output(m, x)
207207
return x, attn_data
208208

209-
class FeedFowardNetwork(nn.Module):
209+
class FeedForwardNetwork(nn.Module):
210210
"""
211211
Position-wise feed-forward networks succeeding the attention layer
212212
@@ -269,7 +269,7 @@ def __init__(self,
269269
dropout_prob=dropout_prob,
270270
return_weights=return_weights
271271
)
272-
self.ffn = FeedFowardNetwork(
272+
self.ffn = FeedForwardNetwork(
273273
hidden_size=hidden_size,
274274
intermediate_size=intermediate_size,
275275
dropout_prob=dropout_prob

0 commit comments

Comments
 (0)