nayeemrizve / ups

"In Defense of Pseudo-Labeling: An Uncertainty-Aware Pseudo-label Selection Framework for Semi-Supervised Learning" by Mamshad Nayeem Rizve, Kevin Duarte, Yogesh S Rawat, Mubarak Shah (ICLR 2021)
MIT License
231 stars 40 forks source link

IndexError: arrays used as indices must be of integer (or boolean) type #13

Open BianTJ opened 1 year ago

BianTJ commented 1 year ago

When I run the source code, it gives me the error:

pseudo_target = pseudo_target[blnc_idx_list] IndexError: arrays used as indices must be of integer (or boolean) type

The error occurs in the pseudo_labeling_util.py. image

After debugging, the variable out_std and out_std_nl were found be torch.tnsor([nan,nan,])

How can I fix this code error?