main#
- main()#
Main function to run the experiment.
This function orchestrates the entire experiment lifecycle, including: 1. Parsing command-line arguments. 2. Loading and overriding configuration from files and command line. 3. Setting up the environment, including seeding for reproducibility and configuring torch settings. 4. Instantiating the data module, network model, and the Lightning wrapper. 5. Setting up the Weights & Biases logger, with support for auto-resuming runs. 6. Handling checkpoint loading for starting from pretrained weights or fine-tuning. 7. Constructing the PyTorch Lightning trainer with appropriate callbacks. 8. Executing the training, validation, and testing phases of the experiment.
- Return type:
None