From 169823ce0bb8f758d1324d8724e377c346118b16 Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Wed, 13 Oct 2021 10:49:06 +0000 Subject: [PATCH] remove useless device --- deepspeech/decoders/scores/ctc_prefix_score.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/deepspeech/decoders/scores/ctc_prefix_score.py b/deepspeech/decoders/scores/ctc_prefix_score.py index 77ac09cd..754e43ae 100644 --- a/deepspeech/decoders/scores/ctc_prefix_score.py +++ b/deepspeech/decoders/scores/ctc_prefix_score.py @@ -40,7 +40,6 @@ class CTCPrefixScorePD(): self.input_length = x.size(1) self.odim = x.size(2) self.dtype = x.dtype - self.device = x.place # Pad the rest of posteriors in the batch # TODO(takaaki-hori): need a better way without for-loops @@ -62,7 +61,7 @@ class CTCPrefixScorePD(): # B idx, hyp idx. shape (B*W, 1) self.idx_bh = None # B idx. shape (B,) - self.idx_b = paddle.arange(self.batch, place=self.device) + self.idx_b = paddle.arange(self.batch) # B idx, O idx. shape (B, 1) self.idx_bo = (self.idx_b * self.odim).unsqueeze(1)