Fix minor bug in ssl_dataset.py

resolve LeeDoYup#11 .
stablegradients · Jan 28, 2021 · cff901f · cff901f
1 parent 0e0b492
commit cff901f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datasets/ssl_dataset.py b/datasets/ssl_dataset.py
@@ -117,7 +117,7 @@ def get_ssl_dset(self, num_labels, index=None, include_lb_to_ulb=True,
         lb_dset = BasicDataset(lb_data, lb_targets, num_classes, 
                                transform, False, None, onehot)
 
-        ulb_dset = BasicDataset(data, targets, num_classes, 
+        ulb_dset = BasicDataset(ulb_data, ulb_targets, num_classes, 
                                transform, use_strong_transform, strong_transform, onehot)
 
-        return lb_dset, ulb_dset
+        return lb_dset, ulb_dset