I find some suspicious code in decode process

[BboyHanat]

In decode.py line 188:
"mask_2 = (mask_2 == 7).float().expand(batch, num_joints, K, 2)"

but when I debug this code, there are all "False" value, since mask_2 is a bool tensor(also, my pytorch vision is 2.0.0, maybe it's correct in lower vision)

[AnukritiSinghh]

were you able to correct this?