TypeError:can't convert np.ndarray of type numpy.bool_

[menggehe]

Traceback (most recent call last): File "main.py", line 89, in num_predict=args.num_predict) File "/home/hemengge/xlnet-pytorch/data_utils.py", line 335, in make_permute is_masked = torch.Tensor(feature.pop("is_masked"))

TypeError: can't convert np.ndarray of type numpy.bool_. The only supported types are: double, float, float16, int64, int32, and uint8

[LorraineLo4]

Having the same issue.

[xingchensong]

is_masked = torch.ByteTensor(feature.pop("is_masked").copy().astype(np.uint8)) @menggehe