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