Browse Source

small - 修改paddle_padder中dtype的判断条件

tags/v1.0.0alpha
x54-729 3 years ago
parent
commit
368fb46913
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      fastNLP/core/collators/padders/paddle_padder.py

+ 1
- 1
fastNLP/core/collators/padders/paddle_padder.py View File

@@ -56,7 +56,7 @@ def is_paddle_dtype_str(dtype):


def _get_dtype(ele_dtype, dtype, class_name):
if not (ele_dtype is not None or is_number_or_numpy_number(ele_dtype) or is_paddle_tensor(ele_dtype) or is_paddle_dtype_str(ele_dtype)):
if not (ele_dtype is None or is_number_or_numpy_number(ele_dtype) or is_paddle_tensor(ele_dtype) or is_paddle_dtype_str(ele_dtype)):
raise EleDtypeUnsupportedError(f"`{class_name}` only supports padding python numbers "
f"or numpy numbers or paddle.Tensor but get `{ele_dtype}`.")



Loading…
Cancel
Save