fix for plaidML

This commit is contained in:
iperov 2019-04-20 17:29:50 +04:00
parent 8e16f4fa1b
commit dcbdc79e84

View file

@ -301,7 +301,7 @@ def get_plaidML_devices():
return plaidML_devices return plaidML_devices
if not has_nvidia_device: if not has_nvidia_device:
get_plaidML_devices() get_plaidML_devices()
#choosing backend #choosing backend
@ -335,6 +335,7 @@ if force_plaidML or (device.backend is None and not has_nvidia_device):
device.backend = None device.backend = None
else: else:
device.backend = "plaidML" device.backend = "plaidML"
plaidML_devices_count = len(get_plaidML_devices())
if device.backend is None: if device.backend is None:
if force_tf_cpu: if force_tf_cpu: