henry2024 commited on
Commit
b28dd2d
1 Parent(s): 62c3c41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, nonlinearity= 'relu', bias= True)
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):