How and where to save preds.npy and trues.npy after fine-tuning correctly?

[well0203]

Hi,
Thank you for your excellent work! I am wondering where in your code I can add the code to store predictions and true values after fine-tuning in self-supervised learning? It will be also great, if you also provide me with a hint regarding where to use a code for graphs creation (as in self-supervised).