AMP: code refactoring, fix preview history

added dumpdflive command
2021-06-26 10:44:41 +04:00
parent 6d89d7fa4c
commit 5783191849
9 changed files with 143 additions and 144 deletions
@@ -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