Spaces:
Sleeping
Sleeping
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)
|