Is it possible to pass arguments to callbacks in the LightningCLI?

[cgoliver]

For example, if I wanted to set the option for write_interval in the BasePredictionWriter from the LightningCLI's trainer.callbacks flag, would there be a syntax for this? As far as I can tell, one can only pass the name of the callback class.

[mauvilsa]

The syntax is described in https://pytorch-lightning.readthedocs.io/en/stable/cli/lightning_cli_advanced_3.html#trainer-callbacks-and-arguments-with-class-type. As command line arguments would be:

$ python ... \
    --trainer.callbacks+=BasePredictionWriter \
    --trainer.callbacks.write_interval=batch \
    ...

Or in a config file:

trainer:
  callbacks:
    - class_path: BasePredictionWriter
      init_args:
        write_interval: batch