From 8c22c99d66fdeb1c0ebc704677de80398f960814 Mon Sep 17 00:00:00 2001 From: Anastasiia Kutakh Date: Tue, 3 Aug 2021 13:57:25 -0400 Subject: [PATCH] . --- CDTools/models/base.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/CDTools/models/base.py b/CDTools/models/base.py index 5ceca97..0da45eb 100644 --- a/CDTools/models/base.py +++ b/CDTools/models/base.py @@ -42,6 +42,8 @@ import time #import pytorch_warmup from .complex_adam import MyAdam from .complex_lbfgs import MyLBFGS +from matplotlib.backends.backend_pdf import PdfPages +pp = PdfPages('multipage.pdf') __all__ = ['CDIModel'] @@ -514,6 +516,7 @@ class CDIModel(t.nn.Module): if update: plt.draw() fig.canvas.start_event_loop(0.001) + pp.savefig() if first_update: plt.pause(0.05 * len(self.figs))