github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-14 02:37:00 -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
commit 4d37fd62cd
11 changed files with 174 additions and 101 deletions
Download patch file Download diff file Expand all files Collapse all files

2
nnlib/devicelib.py
View file

@ -68,7 +68,7 @@ class devicelib:
@staticmethod
def getDevicesWithAtLeastTotalMemoryGB(totalmemsize_gb):
if not hasNVML and totalmemsize_gb <= 2:
if not hasNVML:
return [0]
result = []

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