Refactor - Githubissues

jscuds / rf-bert

MIT License

0 stars 0 forks source link

Refactor #1

Closed jxmorris12 closed 2 years ago

jxmorris12 commented 2 years ago

separate into files: dataloader, train, models, etc.
set up argparse in train.py
specify an Experiment class to differentiate fine-tuning from retrofitting
set up pytest and move tests into pytest format (just run pytest from root directory)
add requirements.txt
some other small fixes