Fix an issue with wandb not displaying updated config changes
# Bugfix
- Fixes an issue where Wandb will not plot the updated configs even after calling model.cfg = new_cfg.

Closes #3372 

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