Update training_utils.py
When the tensorflow version is r1.2 and lower, the name of list_devices() funtion's output will not have "device:" keyword, because of the PR This change will work just fine.
@@ -9,7 +9,7 @@ def _get_available_devices():
def _normalize_device_name(name):
- name = '/' + name.lower().split('device:')[1]
+ name = '/' + ':'.join(name.lower().replace('/', '').split(':')[-2:])
return name
