MixMatch-pytorch
MixMatch-pytorch copied to clipboard
YU1ut
Unlabelled batch size
According to the paper, there's a set of labelled and unlabelled datapoints that are mixed. In the code nevertheless there are two unlabelled sets, u1 and u2. Both of them are added to the mixed set. So my question is, why are there two such sets?
In Sec. 3.1 of the paper, ''For each u in the batch of unlabeled data U, we generate K augmentations.'' And K is 2 in the paper, so u1 and u2 are used.
In Sec. 3.1 of the paper, ''For each u in the batch of unlabeled data U, we generate K augmentations.'' And K is 2 in the paper, so u1 and u2 are used.
Thank you,another question is that in the code "inputs_u" and "inputs_u2" are from the different batch ? Can I change "(inputs_u, inputs_u2), _ = unlabeled_train_iter.next()" into "inputs_u = unlabeled_train_iter.next() inputs_u2 = unlabeled_train_iter.next()" ? Thank you so much.
In Sec. 3.1 of the paper, ''For each u in the batch of unlabeled data U, we generate K augmentations.'' And K is 2 in the paper, so u1 and u2 are used.
I use another dataset.
They are from the same batch but with different augmentations. See here: https://github.com/YU1ut/MixMatch-pytorch/blob/master/dataset/cifar10.py#L7-L14