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

---
 CDTools/models/base.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/CDTools/models/base.py b/CDTools/models/base.py
index 1b3405d..85c5fbe 100644
--- a/CDTools/models/base.py
+++ b/CDTools/models/base.py
@@ -505,6 +505,7 @@ class CDIModel(t.nn.Module):
                     try:
                         plotter(self, fig, dataset)
                         plt.title(name)
+                        print('f', idx)
                         plt.savefig('img-{0}.pdf'.format(idx), bbox_inches='tight')
                     except (IndexError, KeyError, AttributeError, np.linalg.LinAlgError) as e:
                         pass