Spaces:

ando55
/

clinical_segment_splitter

Runtime error

ando55 commited on Feb 10, 2023

Commit

d38a1be

1 Parent(s): da90a83

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -51,7 +51,7 @@ class PointerNetworks(nn.Module):
         else:
             print('rnn_type should be LSTM,GRU')
-        self.use_cuda = True
         self.nnSELU = nn.SELU()

         else:
             print('rnn_type should be LSTM,GRU')
+        self.use_cuda = False
         self.nnSELU = nn.SELU()