fix

chintu619 · May 12, 2022 · 6e9035d · 6e9035d
1 parent 1c344a9
commit 6e9035d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/espnet/nets/pytorch_backend/nets_utils.py b/espnet/nets/pytorch_backend/nets_utils.py
@@ -153,7 +153,7 @@ 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)
+        assert isinstance(lengths, torch.Tensor), type(lengths)
         lengths = lengths.long()
 
     bs = int(len(lengths))