Closed onionysy closed 3 years ago
When I training the net, the warning raise. [W IndexingUtils.h:20] Warning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead. (function expandTensors) Can you tell me how to fix it ?
@onionysy did you fix this?
I got rid of the warnings by changing line 55 in the criteria.py from
valid_mask = valid_mask.byte().detach()
to
valid_mask = valid_mask.bool().detach()
When I training the net, the warning raise. [W IndexingUtils.h:20] Warning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead. (function expandTensors) Can you tell me how to fix it ?