This commit is contained in:
iperov committed 2019-04-23 17:03:53 +04:00
1 parent 47f9bad42b
commit 268b402513
2 files changed
+3 -3

No files matched your search

+1 -1
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