Input parameters of the training function

To run the code:

1. Install the requirements

2. Train the model

python -m lightning.main --default_root_dir <path_to_save_models> --dataset_root <dataset_root_path> --precision 16 --dataset <dataset_name> --mode tr --batch_size <batch_size> --backbone <name_of_backbone> --accelerator auto --devices 1 --max_epochs <n_epochs> --lr <learning_rate> --weight_decay <weight_decay>

There are other parameters that you can pass. Refer to:

1. lightning/module.py
2. utils/parsers.py