This commit is contained in:
Anastasiia Kutakh
2021-08-04 00:09:54 -04:00
parent 35c1643721
commit 9f580102ba
+1
View File
@@ -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