We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 87a8744 commit 188b060Copy full SHA for 188b060
selene_sdk/predict/model_predict.py
@@ -582,7 +582,7 @@ def get_predictions(self,
582
sequence = self._pad_or_truncate_sequence(input)
583
seq_enc = self.reference_sequence.sequence_to_encoding(sequence)
584
seq_enc = np.expand_dims(seq_enc, axis=0) # add batch size of 1
585
- return predict(self.model, seq_enc)
+ return predict(self.model, seq_enc, use_cuda=self.use_cuda)
586
elif input.endswith('.fa') or input.endswith('.fasta'):
587
self.get_predictions_for_fasta_file(
588
input, output_dir, output_format=output_format)
0 commit comments