train-preedict: close matplotlib after we've finished

they act like file handles
This commit is contained in:
Starbeamrainbowlabs 2022-10-21 15:19:31 +01:00
parent 81e53efd9c
commit 0cb2de5d06
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -43,3 +43,4 @@ def vis_embeddings(filepath_output, features):
plt.suptitle(f"RainfallContrastive embeddings | rainfall | E2 ConvNeXt | {len(features)} items", fontsize=28, weight="bold") plt.suptitle(f"RainfallContrastive embeddings | rainfall | E2 ConvNeXt | {len(features)} items", fontsize=28, weight="bold")
plt.savefig(filepath_output) plt.savefig(filepath_output)
plt.close()