Bhuvanesh24 commited on
Commit ·
6743bb5
1
Parent(s): d513c2a
recovered error
Browse files- src/model.py +1 -1
src/model.py
CHANGED
|
@@ -3,7 +3,7 @@ import torch.nn as nn
|
|
| 3 |
import math
|
| 4 |
|
| 5 |
class EnhancedLSTM(nn.Module):
|
| 6 |
-
def
|
| 7 |
super(EnhancedLSTM, self)._init_()
|
| 8 |
self.input_size = input_size
|
| 9 |
|
|
|
|
| 3 |
import math
|
| 4 |
|
| 5 |
class EnhancedLSTM(nn.Module):
|
| 6 |
+
def __init__(self, input_size, lstm_layer_sizes, linear_layer_size, output_size):
|
| 7 |
super(EnhancedLSTM, self)._init_()
|
| 8 |
self.input_size = input_size
|
| 9 |
|