selected_label

[lzw1997lzw]

Hello, I read the code you open source to Github and found that the size of selected_label is the size of the entire unlabeled dataset. When updating selected_label later, the index used is the current batch size index, for example, batch is 64, X_ULb_idx is the index range of 0-63. Or the index of the original dataset, I don't know if you can understand this description?

[Hhhhhhao]

The index 'idx_ulb' used to update selected_label is the index in original dataset, not the batch index. You can check dataset code to verify this:

TorchSSL/datasets/dataset.py

Line 87 in f2f4607

return idx, img_w, self.strong_transform(img)