Contains all the Python scrips required to: create, train/test, evaluate and predict on Pytorch models.
The files will be updated with more useful functions. This about all the "main" functions I use for testing models.
To download into your environment using pip, run:
pip install "git+https://github.com/Dristro/nn_scripts.git"
Make sure to have GIT installed for cloning the repo