Give error if passing the wrong type (e.g., a module) to checkpoint save or restore

[carlini]

Currently if someone accidentally calls save or restore on a module, things crash with an unhelpful error. This gives a better error.