david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix saehd for multi gpu

Browse Source
This commit is contained in:
Colombo
2020-02-27 23:06:06 +04:00
parent 74999ce7ee
commit f88c45d338
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/Model_SAEHD/Model.py
+3 -2
View File
@@ -621,8 +621,9 @@ class SAEHDModel(ModelBase):
pred_src_srcm = nn.tf_concat(gpu_pred_src_srcm_list, 0) pred_src_srcm = nn.tf_concat(gpu_pred_src_srcm_list, 0)
pred_dst_dstm = nn.tf_concat(gpu_pred_dst_dstm_list, 0) pred_dst_dstm = nn.tf_concat(gpu_pred_dst_dstm_list, 0)
pred_src_dstm = nn.tf_concat(gpu_pred_src_dstm_list, 0) pred_src_dstm = nn.tf_concat(gpu_pred_src_dstm_list, 0)
src_loss = nn.tf_average_tensor_list(gpu_src_losses)
dst_loss = nn.tf_average_tensor_list(gpu_dst_losses) src_loss = tf.concat(gpu_src_losses, 0)
dst_loss = tf.concat(gpu_dst_losses, 0)
src_dst_loss_gv_op = self.src_dst_opt.get_update_op (nn.tf_average_gv_list (gpu_G_loss_gvs)) src_dst_loss_gv_op = self.src_dst_opt.get_update_op (nn.tf_average_gv_list (gpu_G_loss_gvs))
if self.options['true_face_power'] != 0: if self.options['true_face_power'] != 0: