github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-16 10:03:41 -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
parent 47f9bad42b
commit 268b402513
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
models/Model_DEV_FANSEG/Model.py
View file

@ -65,9 +65,9 @@ class Model(ModelBase):
def onTrainOneIter(self, generators_samples, generators_list):
target_src, target_src_mask = generators_samples[0]
loss,acc = self.fan_seg.train_on_batch( [target_src], [target_src_mask] )
loss = self.fan_seg.train_on_batch( [target_src], [target_src_mask] )
return ( ('loss', loss), ('acc',acc))
return ( ('loss', loss), )
#override
def onGetPreview(self, sample):