henry2024 commited on
Commit
44ee36d
1 Parent(s): b28dd2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -63,11 +63,11 @@ 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, bias= True)
67
  self.output= nn.Linear(in_features=240, out_features=24)
68
 
69
  def forward(self, x):
70
- y, hidden= self.gru(x)
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)