Update hyper parameter saving
# Bugfix
- Fix issues with config saving `TypeError: __init__() missing 1 required positional argument: 'cfg'` when trying to do model.load_from_checkpoint(XYZ)

Signed-off-by: smajumdar <titu1994@gmail.com>