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

_

Browse source
This commit is contained in:
iperov 2019-04-23 17:03:53 +04:00
commit 268b402513
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
facelib/FANSegmentator.py
View file

@ -50,7 +50,7 @@ class FANSegmentator(object):
if training:
#self.model.compile(loss='mse', optimizer=Adam(tf_cpu_mode=2))
self.model.compile(loss='binary_crossentropy', optimizer=Adam(tf_cpu_mode=2), metrics=['accuracy'])
self.model.compile(loss='binary_crossentropy', optimizer=Adam(tf_cpu_mode=2) )
def __enter__(self):
return self