手机重度患者
2021-08-09 23:04
采纳率: 100%
浏览 44
已结题

使用pytorch中Resne网络的识别问题

训练完resnet网络之后,想使用resnet直接对单张图片进行分类判别,但因为resnet存在BN层,导致出现报错“Expected more than 1 value per channel when training”,请问有什么方法解决?

  • 好问题 提建议
  • 收藏

2条回答 默认 最新

相关推荐 更多相似问题