Created by: daming-lu

训练没问题，但是会 run 很久，所以没有等 run 完

infer.py 会 crash

我记得 infer 是不会用到 train 的结果的，那不知道为啥会crash。如果需要用到训练完的model，如果不大的话，能否提供在github里？这样能确保 infer 不 crash