minor clean up changes

12341123 · Sep 18, 2021 · cbfe01c · cbfe01c
1 parent 85ba2f5
commit cbfe01c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 11 deletions.
diff --git a/chest.py b/chest.py
@@ -33,7 +33,6 @@
 parser = argparse.ArgumentParser(description='X-RAY Pathology Detection')
 parser.add_argument('--seed', type=int, default=0, help='')
 parser.add_argument('--dataset_dir', type=str, default="./data/")
-parser.add_argument('--ssl_ckpt_dir', type=str, default="")
 parser.add_argument('--lr', type=float, default=0.001, help='')
 parser.add_argument('--weight_decay', type=float, default=1e-5, help='')
 
@@ -217,15 +216,15 @@ def tqdm(*args, **kwargs):
         if train_loader.index(dataloader) == train_datas.index(data):
             tr_l = DataLoader(xrv.datasets.SubsetDataset(dataset=data, idxs=range(cfg.batch_size*cfg.num_batches)),
                            batch_size=cfg.batch_size,
-                           shuffle=True,
+                           shuffle=cfg.shuffle,
                            num_workers=cfg.num_workers,
                            pin_memory=True,
                            drop_last=True)
             dataloader.insert(0, tr_l)
 
 valid_loader = DataLoader(valid_data,
                        batch_size=cfg.batch_size,
-                       shuffle=True,
+                       shuffle=cfg.shuffle,
                        num_workers=cfg.num_workers,
                        pin_memory=True,
                        drop_last=True)

diff --git a/xrv_test.py b/xrv_test.py
@@ -68,14 +68,6 @@ def tqdm(*args, **kwargs):
 
 transforms = torchvision.transforms.Compose([xrv.datasets.XRayCenterCrop(), xrv.datasets.XRayResizer(112)])
 
-#xrv.datasets.default_pathologies = [
-#                             'Cardiomegaly',
-#                             'Effusion',
-#                             'Edema',
-#                             'Consolidation',
-#                                   ]
-
-
 if "nih" in cfg.dataset_name:
     ### Load NIH Dataset ### 
     NIH_dataset = xrv.datasets.NIH_Dataset(