- Training with custom hyperparameters
- Training with tuned hyperparameters (tuning is multi-processed for faster execution)
- Model saving/loading
- Training and evaluation metrics visualization
Install and run with --help option, this will explain usage details.
python3.12 -m venv venv
source venv/bin/activate
pip install -r requirements.txt
python3.12 main.py --help