fixed data dir path

Author: Ubuntu

src/aging_gan/data.py

@@ -12,13 +12,13 @@
 
 class UTKFace(Dataset):
     """
-    Assumes the unzipped UTKFace images live in  <root>/data/UTKFace
+    Assumes the unzipped aligned UTKFace images live in  <root>/data/utkface_aligned_cropped/UTKFace
     File pattern:  {age}_{gender}_{race}_{yyyymmddHHMMSS}.jpg
     """
 
     def __init__(self, root: str, transform: T.Compose | None = None):
         self.root = (
-            Path(root) / "UTKFace"  # "utkface_aligned_cropped" /
+            Path(root) / "utkface_aligned_cropped" / "UTKFace"
         )  # or "UTKFace" for the unaligned and varied original version.
         self.files = sorted(f for f in self.root.glob("*.jpg"))
         if not self.files: