From 9b8ac050de87eef5f88600c70837e50f0aa41203 Mon Sep 17 00:00:00 2001
From: zxcd <228587199@qq.com>
Date: Fri, 2 Jun 2023 14:17:00 +0800
Subject: [PATCH] add dtype param for arange API. (#3302)

---
 .../s2t/models/wav2vec2/processing/speech_augmentation.py      | 3 ++-
 paddlespeech/vector/io/augment.py                              | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/paddlespeech/s2t/models/wav2vec2/processing/speech_augmentation.py b/paddlespeech/s2t/models/wav2vec2/processing/speech_augmentation.py
index 5482ed56..50a95f0b 100644
--- a/paddlespeech/s2t/models/wav2vec2/processing/speech_augmentation.py
+++ b/paddlespeech/s2t/models/wav2vec2/processing/speech_augmentation.py
@@ -354,7 +354,8 @@ class Resample(nn.Layer):
         window_width = self.lowpass_filter_width / (2.0 * lowpass_cutoff)
 
         assert lowpass_cutoff < min(self.orig_freq, self.new_freq) / 2
-        output_t = paddle.arange(start=0.0, end=self.output_samples)
+        output_t = paddle.arange(
+            start=0.0, end=self.output_samples, dtype='int64')
         output_t /= self.new_freq
         min_t = output_t - window_width
         max_t = output_t + window_width
diff --git a/paddlespeech/vector/io/augment.py b/paddlespeech/vector/io/augment.py
index 0aa89c6a..ffda1386 100644
--- a/paddlespeech/vector/io/augment.py
+++ b/paddlespeech/vector/io/augment.py
@@ -343,7 +343,8 @@ class Resample(nn.Layer):
         window_width = self.lowpass_filter_width / (2.0 * lowpass_cutoff)
 
         assert lowpass_cutoff < min(self.orig_freq, self.new_freq) / 2
-        output_t = paddle.arange(start=0.0, end=self.output_samples)
+        output_t = paddle.arange(
+            start=0.0, end=self.output_samples, dtype='int64')
         output_t /= self.new_freq
         min_t = output_t - window_width
         max_t = output_t + window_width