Check if covariance and mapping keys match when loading state dict

Check if covariance and mapping keys match when loading state dict #617