Method distribution_alignment

models/softmatch/softmatch.py:263–267

(self, probs)

    @torch.no_grad()
    def distribution_alignment(self, probs):
        # da
        probs = probs * self.lb_prob_t / self.ulb_prob_t
        probs = probs / probs.sum(dim=1, keepdim=True)
        return probs.detach()


    @torch.no_grad()