fangchangma / self-supervised-depth-completion

ICRA 2019 "Self-supervised Sparse-to-Dense: Self-supervised Depth Completion from LiDAR and Monocular Camera"
MIT License
623 stars 135 forks source link

To much warning. #53

Closed onionysy closed 3 years ago

onionysy commented 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 ?

FrederikWarburg commented 3 years ago

@onionysy did you fix this?

FrederikWarburg commented 3 years ago

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()