Zhiqi Li
Zhiqi Li
I meet same issue and have solved it. If you ensure that you have install cuda driver and the version of cuda and torch are matched, and your cuda install...
Curious about this, too.
>  I wonder why pred_normal[..., 0] = 1 - pred_normal[..., 0] here Same question, have you made it clear?