From 26f3049f4cdd07061efa5921074a567dd2291961 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Tue, 6 Jun 2023 15:30:14 +0800 Subject: [PATCH] minor fixes --- egs/librispeech/ASR/zipformer/zipformer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/egs/librispeech/ASR/zipformer/zipformer.py b/egs/librispeech/ASR/zipformer/zipformer.py index 7dd794e0e..2cfc29e49 100644 --- a/egs/librispeech/ASR/zipformer/zipformer.py +++ b/egs/librispeech/ASR/zipformer/zipformer.py @@ -1321,8 +1321,6 @@ class CompactRelPositionalEncoding(torch.nn.Module): # self.pe contains both positive and negative parts # the length of self.pe is 2 * input_len - 1 if self.pe.size(0) >= T * 2 - 1: - # Note: TorchScript doesn't implement operator== for torch.Device - # if self.pe.dtype != x.dtype or self.pe.device != x.device: self.pe = self.pe.to(dtype=x.dtype, device=x.device) return