RabiaSufian commited on
Commit
331b272
·
verified ·
1 Parent(s): b32fda0

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -2,7 +2,7 @@ import torch.nn as nn
2
  import torch
3
 
4
  class LSTMClassifier(nn.Module):
5
- def __init__(self, input_size=1, hidden_size=32, num_layers=1,
6
  bidirectional=True, dropout=0.0, num_classes=2):
7
  super(LSTMClassifier, self).__init__()
8
  self.hidden_size = hidden_size
 
2
  import torch
3
 
4
  class LSTMClassifier(nn.Module):
5
+ def __init__(self, input_size=1, hidden_size=64, num_layers=1,
6
  bidirectional=True, dropout=0.0, num_classes=2):
7
  super(LSTMClassifier, self).__init__()
8
  self.hidden_size = hidden_size