Closed njallskarp closed 1 year ago
This is the file that will be run when we call the training. It will give us the chance to run the training with a command line call like this:
python main.py --lr 1e-3 --batch_size 16
This is just the boilerplate for parsing the basic arguments for such a training pipeline
poetry run ruff .
Fixed some things in the metrics.py file to pass new ruff linter rules
Main File
This is the file that will be run when we call the training. It will give us the chance to run the training with a command line call like this:
python main.py --lr 1e-3 --batch_size 16
This is just the boilerplate for parsing the basic arguments for such a training pipeline
testing
poetry run ruff .
Other stuff
Fixed some things in the metrics.py file to pass new ruff linter rules