Created by: peterzhang2029
This is a follow-up work of https://github.com/PaddlePaddle/models/pull/63