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