Had an error about multiplying Bool with ByteTensor

[mitalbert]

The training script would stop throwing an error about multiplying Bool with ByteTensor in segmentation_miou.py at

        pred = pred * (target>0)
        inter = pred * (pred == target)

I assume the authors were expecting torch to cast Bool values into 0 and 1, which didn't happen.
I changed it to

        pred = pred * (target>0).type(torch.ByteTensor)
        inter = pred * .type(torch.ByteTensor)

to explicitly cast the values.

I'm not sure why this happened in my case, perhaps due to version incompatibility.

[sacmehta]

This is due to version incompatibility. Could you please share your PyTorch version so that we can debug more? Thanks

[mitalbert]

Sure, PyTorch '1.2.0'. I wonder why they committed this useful functionality.

[sacmehta]

There are some more issues in v1.2. I would recommend to use v1.1.

[mitalbert]

Thanks and agreed, had few issues after upgrading.