Spaces:

ando55
/

clinical_segment_splitter

Runtime error

ando55 commited on Feb 10, 2023

Commit

68a4358

1 Parent(s): 940e935

Update solver.py

Files changed (1) hide show

solver.py CHANGED Viewed

@@ -82,8 +82,6 @@ def sample_a_sorted_batch_from_numpy(numpyX,numpyY,batch_size,use_cuda):
     batch_y = align_variable_numpy(batch_y,maxL,2)
     batch_x = Variable(torch.from_numpy(np.array(batch_x, dtype="int64")))
-    if use_cuda:
-        batch_x = batch_x.cuda()
     return  numpy_batch_x,batch_x,batch_y,index_decoder_X,index_decoder_Y,all_lens,maxL

     batch_y = align_variable_numpy(batch_y,maxL,2)
     batch_x = Variable(torch.from_numpy(np.array(batch_x, dtype="int64")))
     return  numpy_batch_x,batch_x,batch_y,index_decoder_X,index_decoder_Y,all_lens,maxL