not handling gpu mem usage correctly #11
parent
2d3182c9be
commit
582d8e5fd7
|
@ -231,7 +231,7 @@ class CustomCollector(object):
|
||||||
try:
|
try:
|
||||||
for gpu_name, gpu_stats in stats['gpu_usages'].items():
|
for gpu_name, gpu_stats in stats['gpu_usages'].items():
|
||||||
add_metric(gpu_usages, [gpu_name], gpu_stats, 'gpu')
|
add_metric(gpu_usages, [gpu_name], gpu_stats, 'gpu')
|
||||||
add_metric(gpu_usages, [gpu_name], gpu_stats, 'mem')
|
add_metric(gpu_mem_usages, [gpu_name], gpu_stats, 'mem')
|
||||||
except KeyError:
|
except KeyError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue