前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
Created by: lkevinzc
Fix a bug when training with image_shape [1, x, x]