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

.

Browse source
This commit is contained in:
iperov 2020-12-11 17:56:39 +04:00
parent 704b5dc072
commit 1ee7798fb9
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_XSeg/Model.py
View file

@ -101,7 +101,7 @@ class XSegModel(ModelBase):
# Average losses and gradients, and create optimizer update ops
with tf.device(f'/CPU:0'):
with tf.device(f'/CPU:0'): # Temporary fix. Unknown bug with training freeze starts from 2.4.0, but 2.3.1 was ok
#with tf.device (models_opt_device):
pred = tf.concat(gpu_pred_list, 0)
loss = tf.concat(gpu_losses, 0)