Skip to content

S
seminar-breakout
Commit 30248bcf authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

rename GPUUtil statistics for better tensorboard grouping

parent af61ebbc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/callbacks/prof.py
View file @ 30248bcf
...@@ -67,7 +67,7 @@ class GPUUtilizationTracker(Callback): ...@@ -67,7 +67,7 @@ class GPUUtilizationTracker(Callback):
self._evt.set() self._evt.set()
stats = self._queue.get() stats = self._queue.get()
for idx, dev in enumerate(self._devices): for idx, dev in enumerate(self._devices):
self.trainer.monitors.put_scalar('GPU{}-Util'.format(dev), stats[idx]) self.trainer.monitors.put_scalar('GPUUtil/{}'.format(dev), stats[idx])
def _after_train(self): def _after_train(self):
self._stop_evt.set() self._stop_evt.set()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment