github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-10 23:33:30 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

3
nnlib/device.py
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: