main

lightning_ir.main.main()

generate config using python main.py fit –print_config > config.yaml additional callbacks at: https://lightning.ai/docs/pytorch/stable/api_references.html#callbacks

Example

To obtain a default config:

python main.py fit –trainer.callbacks=ModelCheckpoint –optimizer AdamW –trainer.logger LightningIRWandbLogger –print_config > default.yaml

To run with the default config:

python main.py fit –config default.yaml