RuntimeError: result type Byte can't be cast to the desired output type Bool

[asadnorouzi]

I am getting the following runtime error:

RuntimeError: result type Byte can't be cast to the desired output type Bool

It is caused by line 130 in box_coder_utils.py.

[zwqnju]

I just change dir_labels.byte() to dir_labels.bool() and the error is eliminated.

[asadnorouzi]

Thank you @zwqnju. It fixed the error.

[zwqnju]

@asadnorouzi I find that using PyTorch 1.1 can also fix this issue, which do not need modifying code. Higher version of PyTorch does not support the code.

[asadnorouzi]

@zwqnju Yeah, I figured that this repo has been developed based on Pytorch 1.1. I'm testing it with Pytorch 1.4. It's been OK so far.