change the dataset

ZERONE182 · May 19, 2023 · 461a2e8 · 461a2e8
1 parent 5748021
commit 461a2e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GAN/pi-GAN/datasets.py b/GAN/pi-GAN/datasets.py
@@ -80,7 +80,7 @@ def __init__(self,  dataset_path, img_size, **kwargs):
         super().__init__()
         paths = glob.glob(dataset_path)
         self.data = []
-        for path in paths[0:3]:
+        for path in paths:
             self.data = self.data + glob.glob(path + '/rgb/*.png')
         assert len(self.data) > 0, "Can't find data; make sure you specify the path to your dataset"
         self.transform = transforms.Compose(