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.

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

推荐阅读更多精彩内容

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