- final updates before deployment, have to check training initialization.