Browse Source

small

tags/v1.0.0beta
x54-729 2 years ago
parent
commit
e23eb1b6e9
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      fastNLP/core/utils/seq_len_to_mask.py

+ 1
- 1
fastNLP/core/utils/seq_len_to_mask.py View File

@@ -94,4 +94,4 @@ def seq_len_to_mask(seq_len, max_len: Optional[int]=None):
pass

raise TypeError("seq_len_to_mask function only supports numpy.ndarray, torch.Tensor, paddle.Tensor, "
f"and jittor.Var, but got {type(seq_len)}")
f"jittor.Var and oneflow.Tensor, but got {type(seq_len)}")

Loading…
Cancel
Save