can run training and save callbacks

run.py

src/helpers.py

src/modules/model_setup.py

src/modules/training.py

test/test_helpers.py

test/test_modules/test_pre_processing.py