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

.

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)