Skip to content

S
seminar-breakout
Commit 9c5b3ee1 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix bug in 6e0b50

parent 0744f02b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/callbacks/prof.py
View file @ 9c5b3ee1
...@@ -82,7 +82,7 @@ class GPUUtilizationTracker(Callback): ...@@ -82,7 +82,7 @@ class GPUUtilizationTracker(Callback):
else: else:
raise RuntimeError("GPUUtilization.worker() process is killed unexpectedly.") raise RuntimeError("GPUUtilization.worker() process is killed unexpectedly.")
if stats == -1: if isinstance(stats, int) and stats == -1:
from ..train.base import StopTraining from ..train.base import StopTraining
raise StopTraining("GPUUtilizationTracker.worker has failed.") raise StopTraining("GPUUtilizationTracker.worker has failed.")
for idx, dev in enumerate(self._devices): for idx, dev in enumerate(self._devices):
......
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