github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

AMP: code refactoring, fix preview history

Browse source 
added dumpdflive command
This commit is contained in:
iperov 2021-06-26 10:44:41 +04:00
parent 6d89d7fa4c
commit 5783191849
9 changed files with 143 additions and 144 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_XSeg/Model.py
View file

@ -164,7 +164,7 @@ class XSegModel(ModelBase):
return ( ('loss', np.mean(loss) ), )
#override
def onGetPreview(self, samples):
def onGetPreview(self, samples, for_history=False):
n_samples = min(4, self.get_batch_size(), 800 // self.resolution )
srcdst_samples, src_samples, dst_samples = samples