Update dataset/datasets.py

dataset/datasets.py

@@ -42,10 +42,13 @@ class TrainDataset(torch.utils.data.Dataset):
         hint = torch.cat((bw_img * mask, mask), 0)
 
         return bw_img, color_img, hint
-
+        
 class FineTuningDataset(torch.utils.data.Dataset):
     def __init__(self, data_path, transform=None, mult_amount=1):
         self.data = [x for x in os.listdir(os.path.join(data_path, 'real_manga')) if x.endswith('_dfm.png')]
+        print("Files in FineTuning Dataset:", self.data)  # Debug print
+        # rest of your code remains unchanged
+
         self.data_path = data_path
         self.transform = transform
         self.mult_amount = mult_amount