Skip to content

Commit

Permalink
sort stats
Browse files Browse the repository at this point in the history
  • Loading branch information
GreenWizard2015 committed Jul 8, 2024
1 parent 6d66496 commit 68802d8
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions scripts/preprocess-remote.py
Original file line number Diff line number Diff line change
Expand Up @@ -236,6 +236,9 @@ def main(args):
continue
print('Total: %d training frames, %d testing frames' % (trainFrames, testFrames))

# sort each list in stats to preserve the order between runs
for k, v in stats.items():
stats[k] = sorted(v)
# save the stats
with open(os.path.join(folder, 'stats.json'), 'w') as f:
json.dump(stats, f, indent=2)
Expand Down

0 comments on commit 68802d8

Please sign in to comment.