See https://github.com/k2-fsa/icefall/pull/1058