The prototype of ML models.
The project is implemented based on Torch Lightning.
The config file is in ml_prototype/config/transformer_lm.yaml.
Run the training job with the following command:
python ml_prototype/cli.py fit --config ml_prototype/config/transformer_lm.yaml