Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -63,7 +63,7 @@ class GRU_model(nn.Module):
|
|
63 |
def __init__(self):
|
64 |
super().__init__()
|
65 |
|
66 |
-
self.gru= nn.GRU(input_size=1080, hidden_size=240,num_layers=1,
|
67 |
self.output= nn.Linear(in_features=240, out_features=24)
|
68 |
|
69 |
def forward(self, x):
|
|
|
63 |
def __init__(self):
|
64 |
super().__init__()
|
65 |
|
66 |
+
self.gru= 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):
|