deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Sort data items in descending order, which I suspect will improve performance because we will hit GC less

Browse Source
This commit is contained in:
James Betker 2022-01-23 19:05:32 -07:00
parent d18aec793a
commit 62475005e4
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/trainer/ExtensibleTrainer.py
View File

@ -185,7 +185,7 @@ class ExtensibleTrainer(BaseModel):
sort_key = opt_get(self.opt, ['train', 'sort_key'], None)
if sort_key is not None:
sort_indices = torch.sort(data[sort_key]).indices
sort_indices = torch.sort(data[sort_key], descending=True).indices
else:
sort_indices = None