码迷,mamicode.com
首页 > 其他好文 > 详细

迁移学习调用预训练模型出现Given input size: (512x3x3). Calculated output size: (512x-3x-3)错误

时间:2020-03-28 15:03:12      阅读:590      评论:0      收藏:0      [点我收藏+]

标签:tps   sel   issues   generic   pool   time   预训练模型   二层   average   

    调用预训练模型ResNet18时报错:RuntimeError: Given input size: (512x3x3). Calculated output size: (512x-3x-3). 
Output size is too small at /opt/conda/conda-bld/pytorch_1535491974311/work/aten/src/THCUNN/generic/
SpatialAveragePooling.cu:
63

其解决方法是将倒数第二层的self.avgpool = nn.AdaptiveAvgPool2d((1, 1))替换为
self.avgpool = nn.AdaptiveAvgPool2d(1)即可。具体原因可能是一个bug.

参考:https://github.com/pytorch/vision/issues/696

迁移学习调用预训练模型出现Given input size: (512x3x3). Calculated output size: (512x-3x-3)错误

标签:tps   sel   issues   generic   pool   time   预训练模型   二层   average   

原文地址:https://www.cnblogs.com/dyclown/p/12587390.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!