not handling gpu mem usage correctly #11
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user