mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix inverted logic

Browse Source
This commit is contained in:
James Betker 2022-05-17 15:39:07 -06:00
parent 519151d83f
commit 6a2c29f596
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/util.py
View File

@ -597,7 +597,7 @@ def load_paths_from_cache(paths, cache_path, exclusion_list=[], endswith=None):
print(f"Excluded {before-len(output)} files.")
if endswith is not None:
before = len(output)
output = list(filter(lambda p: not p.endswith(endswith), output))
output = list(filter(lambda p: p.endswith(endswith), output))
print(f"Excluded {before-len(output)} files with endswith mask. For total of {len(output)} files")
print("Done.")
torch.save(output, cache_path)