Merge branch 'master' of http://developer.hpccube.com/codes/modelzoo/taming-transformers_pytorch

f56ae884 · mashun1 · 031cd626 · 41973afa · f56ae884
Commit f56ae884 authored Mar 01, 2024 by mashun1
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

taming/data/faceshq.py taming/data/faceshq.py +2 -2

No files found.
--- a/taming/data/faceshq.py
+++ b/taming/data/faceshq.py
@@ -137,8 +137,8 @@ class FacesHQValidation(Dataset):
                                                      additional_targets={"coord": "image"})

    def __len__(self):
-        # return len(self.data)
-        return 1000
+        return len(self.data)
+        # return 1000

    def __getitem__(self, i):
        ex, y = self.data[i]