deep coice 3: use np.int64 explicitly

This commit is contained in:
chenfeiyu 2020-06-02 07:07:10 +00:00
parent 31dbc87128
commit 9ba26facf0
1 changed files with 1 additions and 1 deletions

View File

@ -121,7 +121,7 @@ class Evaluator(object):
en.text_to_sequence(
text, p=self.p_replace), dtype=np.int64)
length = len(text)
text_positions = np.arange(1, 1 + length)
text_positions = np.arange(1, 1 + length, dtype=np.int64)
text = np.expand_dims(text, 0)
text_positions = np.expand_dims(text_positions, 0)