Fix config.yml dump in training runs.

Created by: stephenroller

Patch Description Due to changes of logic around NFS handling, config.yml is no longer being saved along with training runs. Small hack to fix that.

Testing steps Launched a 1 node run, validated config.yml exists.