fixed domainnet dataset config

Author: Lisa Dunlap

data_configs/domain_net.yaml

@@ -0,0 +1,3 @@
+DATA:
+  SOURCE_DOMAIN: clipart # options are clipart, infograph, painting, quickdraw, real, sketch
+  TARGET_DOMAIN: real # options are clipart, infograph, painting, quickdraw, real, sketch

helpers/data_helpers.py

@@ -26,6 +26,8 @@ def get_config(name="Waterbirds"):
         cfg       = OmegaConf.load('data_configs/waterbirds.yaml')
     elif "ColoredMNIST" in name:
         cfg       = OmegaConf.load('data_configs/colored_mnist.yaml')
+    elif "DomainNet" in name:
+        cfg       = OmegaConf.load('data_configs/domain_net.yaml')
     else:
         raise ValueError(f"{name} Dataset config not found")
     args      = OmegaConf.merge(base_cfg, cfg)

methods/clip_transformations.py

@@ -186,8 +186,6 @@ def train_debias(self, inputs, labels, groups, dom_gt, test_inputs, test_labels,
                 if self.cfg.EXP.CHECKPOINT_VAL:
                     self.train_val_loop(self.test_loader, epoch, phase="val")
             self.save_checkpoint(0.0, epoch, last=True)
-            # wandb.ss['best val balanced acc'] = self.best_acc
-            # wandb.ss['best epoch'] = self.best_epoch
 
     def load_checkpoint(self, path):
         checkpoint = torch.load(path)