github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

added AMD/Intel cards support via DirectX12 ( DirectML backend )

Browse source
This commit is contained in:
iperov 2021-04-22 18:19:15 +04:00
parent fc4a49c3e7
commit fdb143ff47
7 changed files with 166 additions and 116 deletions
Download patch file Download diff file Expand all files Collapse all files

4
facelib/XSegNet.py
View file

@ -39,7 +39,7 @@ class XSegNet(object):
self.target_t = tf.placeholder (nn.floatx, nn.get4Dshape(resolution,resolution,1) )
# Initializing model classes
with tf.device ('/CPU:0' if place_model_on_cpu else '/GPU:0'):
with tf.device ('/CPU:0' if place_model_on_cpu else nn.tf_default_device_name):
self.model = nn.XSeg(3, 32, 1, name=name)
self.model_weights = self.model.get_weights()
if training:
@ -53,7 +53,7 @@ class XSegNet(object):
self.model_filename_list += [ [self.model, f'{model_name}.npy'] ]
if not training:
with tf.device ('/CPU:0' if run_on_cpu else '/GPU:0'):
with tf.device ('/CPU:0' if run_on_cpu else nn.tf_default_device_name):
_, pred = self.model(self.input_t)
def net_run(input_np):