Hi @xiaoyu258, Good work BTW, I think the model fitting code (to refine the flow map) isn't in this repo. And that's why people can't reproduce your work. Can you update the model fitting code? 