.

2026-09-09 21:12:42 +02:00 · 2021-08-04 09:39:49 -04:00
parent 9f580102ba
commit daeac7e983
1 changed files with 3 additions and 1 deletions
@@ -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: