Merge branch 'xuan_dev' into 'develop'

DTK-203 See merge request dcutoolkit/deeplearing/NNI!3

Merge branch 'xuan_dev' into 'develop'
DTK-203 See merge request dcutoolkit/deeplearing/NNI!3
602e1842 · huchen · fefee175 · 30ac210e · 602e1842
Commit 602e1842 authored Sep 25, 2020 by huchen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/nni_trial_tool/gpu.py tools/nni_trial_tool/gpu.py +1 -1

No files found.
--- a/tools/nni_trial_tool/gpu.py
+++ b/tools/nni_trial_tool/gpu.py
@@ -31,7 +31,7 @@ def parse_nvidia_smi_result(smi):
        output["gpuInfos"] = []
        for gpuIndex, gpu in enumerate(gpuList):
            if gpu == 'system':  
-              break
+              continue
            gpuInfo = {}
            gpuInfo['index'] = gpuIndex
            gpuInfo['gpuUtil'] = smi[gpu]["GPU OverDrive value (%)"]