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

fix xseg

Browse Source
This commit is contained in:
iperov
2020-12-11 20:25:30 +04:00
parent bcfefb492b
commit 26ed582aff
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/Model_XSeg/Model.py
+2 -2
View File
@@ -101,8 +101,8 @@ class XSegModel(ModelBase):
# Average losses and gradients, and create optimizer update ops # Average losses and gradients, and create optimizer update ops
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(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): with tf.device (models_opt_device):
pred = tf.concat(gpu_pred_list, 0) pred = tf.concat(gpu_pred_list, 0)
loss = tf.concat(gpu_losses, 0) loss = tf.concat(gpu_losses, 0)
loss_gv_op = self.model.opt.get_update_op (nn.average_gv_list (gpu_loss_gvs)) loss_gv_op = self.model.opt.get_update_op (nn.average_gv_list (gpu_loss_gvs))