How to run main code:
- Clone the repository.
- Create a new environment from
requirements.txt
. - Run
main.py
.
How to use this as a library:
- You can import functions from from the different files. The code is designed to behave like a module.
- In particular, the function
get_prebuilt_model
is very useful if you want to load in-built (default) models. - You can add new defaults based on configurations that you build.
- It is also possible to build models in a bespoke way.
How to run legacy code:
- The legacy code behaves slightly differently to the main code. It is included for reference.
- In particular,
main_colab.py
andcolab_continue_train.py
are useful if you are running on Colab and have trouble with runtimes restarting. You can use them to save models and then re-train from the previously saved state. main.py
fromlegacy
should not be used.