Checkpoint restore for dynamics model fails without cotraining

[maharajamihir]

Restoring a checkpoint for the dynamics model does not work when cotraining is disabled. The restore process tries to also load the decoder, which causes an error.