Correctly save the model's state_dict

We should try to save the model's state_dict instead of simply saving the object.

This is recommended by PyTorch themselves.

Update https://gitlab.teklia.com/atr/dan/-/blob/ea6286c12784c0d7cb01cf76bade7de3822573ea/dan/ocr/manager/training.py#L440