前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
* use has_grad instead of train_mode * add vlog for debug * fix ut * fix ut