Check for inference checkpoint in checkpoint.py

[havardhhaugen]

It's currently possible to load training checkpoints by mistake, which throws vague errors down the line. We should add a check for this in checkpoint.py when the checkpoint is loaded.

[ways]

I can implement if you tell me what we're looking for.