Saving predictions to file using LightningCLI

[vikigenius]

The predict command in LightningCLI does not specify anything about saving the predictions to file. Are they saved anywhere?

If not how can I extend LightningCLI pedict command to save the predictions, preferably by saving to a specific location.

I see that there is a BasePredictionWriter callback, but I have no idea how to use it or add it to the LightningCLI specifically for the predict subcommand

[tshu-w]

You can ref #14316. Using the callback is a nice idea in the current situation without hacking the CLI directly. To add a callback to CLI:

LightningCLI(
    ...
    trainer_defaults={
        "callbacks": [
            prediction_callback,
        ]
    },
)

or

python cli.py prediction --callbacks+ src.callbacks.prediction_callbacks ...