Spaces:

deepinv
/

denoising

Sleeping

mterris commited on Apr 15

Commit

f71c5d6

1 Parent(s): 9ca4b70

rm ct

Files changed (1) hide show

factories.py CHANGED Viewed

@@ -404,7 +404,7 @@ class BaselineModel(torch.nn.Module):
 class EvalDataset(torch.utils.data.Dataset):
-    all_datasets = ["Natural", "MRI", "CT"]
     def __init__(self, dataset_name: str, device_str: str = "cpu") -> None:
         self.name = dataset_name

 class EvalDataset(torch.utils.data.Dataset):
+    all_datasets = ["Natural", "MRI"]
     def __init__(self, dataset_name: str, device_str: str = "cpu") -> None:
         self.name = dataset_name