mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
fix: always show total last
This commit is contained in:
parent
f50a1f5b0c
commit
4fcfc363af
@ -179,6 +179,9 @@ class ImageLoggingContext:
|
||||
self.summary_context.stop()
|
||||
self.timing_contexts["total"] = self.summary_context
|
||||
|
||||
# move total to the end
|
||||
self.timing_contexts["total"] = self.timing_contexts.pop("total")
|
||||
|
||||
if torch.cuda.is_available():
|
||||
self.summary_context.memory_peak = max(
|
||||
max(context.memory_peak, context.memory_start, context.memory_end)
|
||||
|
Loading…
Reference in New Issue
Block a user