def get_device_name(self, device): """Returns a TF-style device identifier for the specified device. input: 'cpu' for CPU and 'gpuN' for the Nth GPU on the host""" if device == 'cpu': dev_num = 0 dev_type = 'cpu' elif device.startswith('gpu'): try: dev_num = int(device[3:]) dev_type = 'gpu' except ValueError: print ("GPU number could not be parsed from {}; using CPU".format(device)) dev_num = 0 dev_type = 'cpu' else: print ("Please specify 'cpu' or 'gpuN' for device name") dev_num = 0 dev_type = 'cpu' return get_device_name(dev_type, dev_num, backend='tensorflow')