fix

chintu619 · May 12, 2022 · f899a05 · f899a05
1 parent 7d52422
commit f899a05
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/espnet/nets/pytorch_backend/nets_utils.py b/espnet/nets/pytorch_backend/nets_utils.py
@@ -152,6 +152,10 @@ def make_pad_mask(lengths, xs=None, length_dim=-1, maxlen=None):
 
     if not isinstance(lengths, list):
         lengths = lengths.tolist()
+    else:
+        assert isinstance(lengths, torch.tensor), type(lengths)
+        lengths = lengths.long()
+
     bs = int(len(lengths))
     if maxlen is None:
         if xs is None: