To install:
git clone [email protected]:khuddzu/personal_trainer.git
pip install -e .
To initiate the personal trainer protocol
personal_trainer initiate
-
training_center: main directory
-
logs : the directory that all of your trained models will be stored in. Currently the naming format is datetime_projectname.
-
models: the directory that will contain all of your model codes. Currently the model which you want to use must be located in a file called nets.py. Simply change the source if you want this to be different.
-
editor.ini: template configuration file for your networks.
-
template_trainer.py: template code that calls variables from your configuration file and feeds them to protocol trainer, to train your network.
-
template_model_loader.py: template code that loads your trained model, for use.
-
template_nets.py: template code that represents the model class for your network. I use my common ANICHARGE model here.
-
script.sh: template script to send your training job to the queue.
-