Skip to content

Commit

Permalink
Fix comm
Browse files Browse the repository at this point in the history
  • Loading branch information
Javclaude committed May 30, 2020
1 parent be2e1d3 commit 738b008
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/model.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import torch

class LSTMModel(torch.nn.Module):

'''
LSTM model Baseline
Expand All @@ -16,7 +15,6 @@ class LSTMModel(torch.nn.Module):
- n_classes = int, number of classes
- do_normalizarion = bool, if true apply layer normalization
- pooling_strategy = str, pooling over the hidden dimension
'''
def __init__(self,
vocab_size: int,
Expand Down

0 comments on commit 738b008

Please sign in to comment.