Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -63,11 +63,11 @@ class GRU_model(nn.Module):
|
|
| 63 |
def __init__(self):
|
| 64 |
super().__init__()
|
| 65 |
|
| 66 |
-
self.
|
| 67 |
self.output= nn.Linear(in_features=240, out_features=24)
|
| 68 |
|
| 69 |
def forward(self, x):
|
| 70 |
-
y, hidden= self.
|
| 71 |
#print(y.shape)
|
| 72 |
#print(hidden.shape)
|
| 73 |
x= self.output(y)
|
|
|
|
| 63 |
def __init__(self):
|
| 64 |
super().__init__()
|
| 65 |
|
| 66 |
+
self.rnn= nn.GRU(input_size=1080, hidden_size=240, num_layers=1, bias= True)
|
| 67 |
self.output= nn.Linear(in_features=240, out_features=24)
|
| 68 |
|
| 69 |
def forward(self, x):
|
| 70 |
+
y, hidden= self.rnn(x)
|
| 71 |
#print(y.shape)
|
| 72 |
#print(hidden.shape)
|
| 73 |
x= self.output(y)
|