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

revert

Browse Source
This commit is contained in:
iperov
2021-04-19 13:35:14 +04:00
parent af0b3904fc
commit 93fe480eca
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/Model_XSeg/Model.py
-2
View File
@@ -95,8 +95,6 @@ class XSegModel(ModelBase):
gpu_pred_list.append(gpu_pred_t) gpu_pred_list.append(gpu_pred_t)
gpu_loss = tf.reduce_mean( tf.nn.sigmoid_cross_entropy_with_logits(labels=gpu_target_t, logits=gpu_pred_logits_t), axis=[1,2,3]) gpu_loss = tf.reduce_mean( tf.nn.sigmoid_cross_entropy_with_logits(labels=gpu_target_t, logits=gpu_pred_logits_t), axis=[1,2,3])
gpu_loss += tf.reduce_mean ( 0.1*nn.dssim(gpu_target_t, gpu_pred_logits_t, max_val=1.0, filter_size=3), axis=[1])
gpu_losses += [gpu_loss] gpu_losses += [gpu_loss]