HAMA-681: Add function for trainer to load model