github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 21:42:08 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

5
models/Model_SAEHD/Model.py
View file

@ -621,8 +621,9 @@ class SAEHDModel(ModelBase):
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_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))
if self.options['true_face_power'] != 0: