AMP: code refactoring, fix preview history

added dumpdflive command
This commit is contained in:
iperov committed 2021-06-26 10:44:41 +04:00
1 parent 6d89d7fa4c
commit 5783191849
9 files changed
+143 -144

No files matched your search

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