File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -208,12 +208,12 @@ def log_stats(self):
208208
209209 total_time = 0
210210 logger .info (f"Graph stats: { graph_id } " )
211- logger .info ("Node Calls Seconds VRAM Used" )
211+ logger .info ("Node Calls Seconds VRAM Used" )
212212 for node_type , stats in self ._stats [graph_id ].nodes .items ():
213- logger .info (f"{ node_type :<20} { stats .calls :>5} { stats .time_used :4 .3f} s { stats .max_vram :4.2f} G" )
213+ logger .info (f"{ node_type :<20} { stats .calls :>5} { stats .time_used :7 .3f} s { stats .max_vram :4.2f} G" )
214214 total_time += stats .time_used
215215
216- logger .info (f"TOTAL GRAPH EXECUTION TIME: { total_time :4 .3f} s" )
216+ logger .info (f"TOTAL GRAPH EXECUTION TIME: { total_time :7 .3f} s" )
217217 if torch .cuda .is_available ():
218218 logger .info ("Current VRAM utilization " + "%4.2fG" % (torch .cuda .memory_allocated () / 1e9 ))
219219
You can’t perform that action at this time.
0 commit comments