Spaces:
Runtime error
Runtime error
Update solver.py
Browse files
solver.py
CHANGED
@@ -82,8 +82,6 @@ def sample_a_sorted_batch_from_numpy(numpyX,numpyY,batch_size,use_cuda):
|
|
82 |
batch_y = align_variable_numpy(batch_y,maxL,2)
|
83 |
batch_x = Variable(torch.from_numpy(np.array(batch_x, dtype="int64")))
|
84 |
|
85 |
-
if use_cuda:
|
86 |
-
batch_x = batch_x.cuda()
|
87 |
|
88 |
return numpy_batch_x,batch_x,batch_y,index_decoder_X,index_decoder_Y,all_lens,maxL
|
89 |
|
|
|
82 |
batch_y = align_variable_numpy(batch_y,maxL,2)
|
83 |
batch_x = Variable(torch.from_numpy(np.array(batch_x, dtype="int64")))
|
84 |
|
|
|
|
|
85 |
|
86 |
return numpy_batch_x,batch_x,batch_y,index_decoder_X,index_decoder_Y,all_lens,maxL
|
87 |
|