microsoft/Semi-supervised-learning

Assertion on pseudolabel

mioam opened this issue · 2 comments

When I use "pseudolabel" which does not require a strong_transform, the assertion appears:

AssertionError: alg pseudolabel requires strong augmentation

I wonder if there is something wrong in

assert self.alg not in ['fullysupervised', 'supervised', 'pseudolabel', 'vat', 'pimodel', 'meanteacher', 'mixmatch', 'refixmatch'], f"alg {self.alg} requires strong augmentation"
, which should be

assert self.alg in ['fullysupervised', 'supervised', 'pseudolabel', 'vat', 'pimodel', 'meanteacher', 'mixmatch',], f"alg {self.alg} requires strong augmentation"

Maybe "refixmatch" also requires a strong augmentation.

Thanks for pointing that out. Would you like to open a PR for making this change?

Stale issue message