fix for plaidML

2019-04-20 17:29:50 +04:00
parent 8e16f4fa1b
commit dcbdc79e84
1 changed files with 2 additions and 1 deletions
@@ -335,6 +335,7 @@ if force_plaidML or (device.backend is None and not has_nvidia_device):
        device.backend = None
    else:
        device.backend = "plaidML"
+        plaidML_devices_count = len(get_plaidML_devices())

 if device.backend is None:
    if force_tf_cpu: