[QOL] Save the training config in the checkpoint dir.

Created by: stephenroller

Patch Description When initiating a training run, dump a copy of the config in the same folder as the checkpoints and logs.

Testing steps Manual testing.