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

fix

Browse source
This commit is contained in:
seranus 2021-12-06 22:38:06 +01:00
commit 6aee9030d2
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_XSeg/Model.py
View file

@ -196,7 +196,7 @@ class XSegModel(ModelBase):
return ( ('loss', np.mean(loss) ), )
#override
def onGetPreview(self, samples, for_history=False):
def onGetPreview(self, samples, for_history=False, filenames=None):
n_samples = min(4, self.get_batch_size(), 800 // self.resolution )
if self.pretrain: