UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead.


内部实则是bool。
原:

    idxs = torch.Tensor(idxs_0.shape[0]).byte()
    idxs[:] = 1
    idxs[idxs_0 < 0] = 0
    idxs[idxs_end >= int(N / args.gps_step)] = 0
    delta_p_gt = delta_p_gt[idxs]
    idxs_0_bias = idxs_0[idxs]
    idxs_end_bis = idxs_end[idxs]

可以看出,实则idxs内部为bool性质。

改后:

    # 加一个.bool()
    idxs = torch.Tensor(idxs_0.shape[0]).byte().bool()
    idxs[:] = 1
    idxs[idxs_0 < 0] = 0
    idxs[idxs_end >= int(N / args.gps_step)] = 0
    delta_p_gt = delta_p_gt[idxs]
    idxs_0_bias = idxs_0[idxs]
    idxs_end_bis = idxs_end[idxs]


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM