From daeac7e983fbbd39e476282a674006c9c85efd03 Mon Sep 17 00:00:00 2001
From: Anastasiia Kutakh <anastkut@mit.edu>
Date: Wed, 4 Aug 2021 09:39:49 -0400
Subject: [PATCH] .

---
 CDTools/models/base.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/CDTools/models/base.py b/CDTools/models/base.py
index 85c5fbe..3c26947 100644
--- a/CDTools/models/base.py
+++ b/CDTools/models/base.py
@@ -499,7 +499,9 @@ class CDIModel(t.nn.Module):
             try:
                 plotter(self,fig)
                 plt.title(name)
-                
+                print('f', idx)
+                plt.savefig('img-{0}.pdf'.format(idx), bbox_inches='tight')
+            
             except TypeError as e:
                 if dataset is not None:
                     try: