前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
* set norm_type to bn while export model * modify code according to review