github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 13:02:15 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix DFLJPG,

Browse source 
SAE: added "rare sample booster"
SAE: pixel loss replaced to smooth transition from DSSIM to PixelLoss in 15k epochs by default
This commit is contained in:
iperov 2019-02-09 18:53:37 +04:00
parent f93b4713a9
commit 4d37fd62cd
11 changed files with 174 additions and 101 deletions
Download patch file Download diff file Expand all files Collapse all files

1
nnlib/nnlib.py
View file

@ -52,6 +52,7 @@ class nnlib(object):
tf = nnlib.tf
tf_sess = nnlib.tf_sess
tf_reduce_mean = tf.reduce_mean # todo tf 12+ = tf.math.reduce_mean
tf_total_variation = tf.image.total_variation
tf_dssim = nnlib.tf_dssim
tf_ssim = nnlib.tf_ssim