reorganize train script + expose through CLI
jeremyjordan opened this issue · comments
What's your idea?
Right now all of our training args are hard-coded in the script. We should expose these through a CLI.
We should also have the lightning module and the training script in separate .py files.
Describe the solution you'd like
Typer is a nice tool for building CLIs.