`main` function for `train_model.py`

ybracke / transnormer

A lexical normalizer for historical spelling variants using a transformer architecture.

GNU General Public License v3.0

6 stars 1 forks source link

Open ybracke opened 1 year ago

ybracke commented 1 year ago

[ ] Move everything currently under if __name__ == "__main__" into a main() function
[ ] CLI arguments: yes or no?
- config and model path as optional
- if nothing is passed: use default path
- model: (a) there is a directory where checkpoints are stored and (b) you need to set a name for the final model
[ ] ~~Logging instead of printing~~