How to write scipy CSR matrix to checkpoint?

nlp
1

I have scipy matrix, I want to write this as a checkpoint.

I am new to Pytorch, please suggest to me the possible way to write scipy matrix to checkpoint.

2

Add this array as a new entry to a dict and save it via torch.save.

3

@ptrblck thank you for the answer. It will work.