How to save to extra info checkpoint?
I have a problem where I preprocess the input data using different encoders from sklearn.preprocessing.
I would like the same encoding to be used during the inference.
It looks natural to save this into the checkpoint. It is straightforward how to do it manually, and it is pretty clear how to do it in catalyst:
    runner.train(
     ...
        checkpoint_data={"encoder": encoder}
    )

But I cannot figure out how to do it in Pytorch Lighting?