CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix for plaidML

Browse Source
This commit is contained in:
iperov
2019-04-20 17:29:50 +04:00
parent 8e16f4fa1b
commit dcbdc79e84
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nnlib/device.py
+2 -1
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: