Problem in Loss Func. Use ICDAR2015 to train to model, but this problem emerged.

[leonrenlang]

File "/workspace/PixelLink-with-pytorch/main.py", line 144, in
main()
File "/workspace/PixelLink-with-pytorch/main.py", line 133, in main
train(config.epoch, iteration, dataloader, my_net, optimizer, optimizer2, device)
File "/workspace/PixelLink-with-pytorch/main.py", line 73, in train
pixel_pos_weights)
File "/workspace/PixelLink-with-pytorch/criterion.py", line 46, in pixel_loss
self.neg_pixel_weight[i] = self.neg_pixel_weight[i] & (neg_pixel_masks[i]==1)
RuntimeError: Expected object of scalar type Byte but got scalar type Bool for argument #2 'other' in call to _th_and

[ligaoyuan-potato]

你好，请问你训练成功了吗？

[Mageayo]

change
self.neg_pixel_weight[i] = self.neg_pixel_weight[i] & (neg_pixel_masks[i]==1)
to
self.neg_pixel_weight[i] = self.neg_pixel_weight[i] & (neg_pixel_masks[i]==1).byte()