bug in ae_loss about cornernet

[zhangming8]

The type of mask is uint8 so anything that is non-zero would be considered as True and can be used as a boolean mask. Here is a small code snippet to demonstrate this behavior:

import torch
a = torch.arange(10)
b = torch.zeros((10, ), dtype=torch.uint8)
b[2] = 1
a[b]
:1: UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead. (Triggered internally at /pytorch/aten/src/ATen/native/IndexingUtils.h:30.)
tensor([2])
b[2] = 2
a[b]
tensor([2])

This became a deprecated behavior after PyTorch had officially introduced boolean type. But this code is no longer maintained so I didn’t update it for the newer version of PyTorch. If you have more questions, please feel free to let me know.

--by Hei Law