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

2022-01-23 19:05:32 -07:00 · 2022-01-23 19:05:32 -07:00 · 62475005e4
commit 62475005e4
parent d18aec793a
1 changed files with 1 additions and 1 deletions
--- a/codes/trainer/ExtensibleTrainer.py
+++ b/codes/trainer/ExtensibleTrainer.py
@ -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