worked on docs and tests, updated some other tests

.gitignore

run.py

src/modules/model_setup.py

src/modules/training.py

test/test_helpers.py

test/test_modules/test_experiment_setup.py

test/test_modules/test_model_setup.py

test/test_modules/test_training.py