Available tasks:
- Computer Vision
- Image classification (cv_classification_base model)
git clone https://github.com/SueGreen/pytorch-lightning-projects-pipeline-template.git
cd pytorch-lightning-projects-pipeline-template
You can optionally modify a configuration file and then run a task with the following command:
python train.py -c "path/to/config/file"
For example, to run a computer vision classification base task, optionally modify "configs/cv_classification_base_config.json" file and then specify the path to it:
python train.py -c "configs/cv_classification_base_config.json"
To see training process, run tensorboard --logdir "checkpoints/<model_name>"
, where model_name is a model from the list.