astart01 commited on
Commit
6977dde
·
verified ·
1 Parent(s): eac90e2

Update perv.py

Browse files
Files changed (1) hide show
  1. perv.py +1 -1
perv.py CHANGED
@@ -33,7 +33,7 @@ def run():
33
  preds = self.pipeline.predict(X)
34
  return self.label_encoder.inverse_transform(preds), time.time() - start_time
35
 
36
- checkpoint = torch.load('models/lstm/model.pt', map_location=torch.device('cpu'))
37
 
38
  class Attention(nn.Module):
39
  def __init__(self, hidden_dim):
 
33
  preds = self.pipeline.predict(X)
34
  return self.label_encoder.inverse_transform(preds), time.time() - start_time
35
 
36
+ checkpoint = torch.load('lstm/model.pt', map_location=torch.device('cpu'))
37
 
38
  class Attention(nn.Module):
39
  def __init__(self, hidden_dim):