Investigate checkpoint issues during training

Traceback and details in https://redmine.teklia.com/issues/4997#note-37