From 35327e6c9e02a3099fea175949b7f06a98d1a65a Mon Sep 17 00:00:00 2001
From: Wenxuan Ou <35703065+wenxuanou@users.noreply.github.com>
Date: Mon, 12 Jul 2021 13:45:41 -0400
Subject: [PATCH] Update solver.py

Since ground truth for labeled samples are ones, that of unlabeled samples should be zeros.
---
 solver.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/solver.py b/solver.py
index 3282ec4..62d0c5c 100644
--- a/solver.py
+++ b/solver.py
@@ -88,7 +88,7 @@ def train(self, querry_dataloader, val_dataloader, task_model, vae, discriminato
                 unlabeled_preds = discriminator(unlab_mu)
                 
                 lab_real_preds = torch.ones(labeled_imgs.size(0))
-                unlab_real_preds = torch.ones(unlabeled_imgs.size(0))
+                unlab_real_preds = torch.zeros(unlabeled_imgs.size(0))
                     
                 if self.args.cuda:
                     lab_real_preds = lab_real_preds.cuda()