Saving the model in HDF5 format