RuntimeError: Given input size: (10x2x2). Calculated output size: (10x-7x-7). Output size is too ...

RuntimeError: Given input size: (10x2x2). Calculated output size: (10x-7x-7). Output size is too small at /opt/conda/conda-bld/pytorch_1535491974311/work/aten/src/THCUNN/generic/

SpatialAveragePooling.cu:63

解决方法：将最后一层 nn.AvgPool2d(class_num， stride=1)

改为：

nn.AvgPool2d(class_num)

解决，这个应该是pytorch1.1的bug.

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容

  • 记PyTorch踩过的坑～(更新中)
    像认真记录生活一样记录Bug. 未经允许，不得转载，谢谢~~ 1. 从autograd.Variable中取Te...
    与阳光共进早餐阅读 60,645评论 17赞 34
  • 深度残差网络理解与Tensorflow/keras/pytorch实现
    欢迎关注本人的微信公众号AI_Engine 各位码神大家好，这是深度学习进阶的第一篇，不要吝啬你们的赞和转发哟~ ...
    AI_Engine阅读 4,327评论 1赞 1
  • 家长会谈合作
    久违的晴天，家长会。 家长大会开好到教室时，离放学已经没多少时间了。班主任说已经安排了三个家长分享经验。 放学铃声...
    飘雪儿5阅读 7,552评论 16赞 22
  • 2019－11－28 周四 阴
    今天感恩节哎，感谢一直在我身边的亲朋好友。感恩相遇！感恩不离不弃。 中午开了第一次的党会，身份的转变要...
    迷月闪星情阅读 10,605评论 0赞 11
  • 赵原野传 国家
    可爱进取，孤独成精。努力飞翔，天堂翱翔。战争美好，孤独进取。胆大飞翔，成就辉煌。努力进取，遥望，和谐家园。可爱游走...
    赵原野阅读 2,760评论 1赞 1