在这里插入图片描述
RuntimeError: Given input size: (256x1x1). Calculated output size: (256x0x0). Output size is too small

1.初始网络模型

原始网络模型对于cifar10使用了六层卷积，且每层卷积后接一层池化。
池化大小为（2*2，stride=2）。

2.报错原因

图片尺寸太小（初始尺寸32323，而32=2^5，经过五层池化后尺寸就变为1），进入网络后池化不够大。

3.解决方法

①修改pooling层，步长变为1。
②减少pooling层，两层卷积后一层池化，池化一共3层。

修改成功，不报错了。

# 深度学习 # 计算机视觉 # 神经网络
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

cover

手把手教你玩转开源大模型：从Llama3到企业级应用

avatar 2048 AI社区

前端老旧项目全面性能优化指南与面试攻略

核心 Web VitalsLCP (Largest Contentful Paint): 最大内容绘制时间FID (First Input Delay): 首次输入延迟CLS (Cumulative Layout Shift): 累积布局偏移其他重要指标浏览器开发者工具Lighthouse 自动化分析WebPageTest 分析加载性能问题运行时性能问题代码分割与懒加载避免阻塞主线程内存优化2.

avatar 2048 AI社区

51c大模型~合集186

9 月 25 日，生数科技新一代图生视频大模型 Vidu Q2 正式全球上线，打破了原有 AI 生成的表情太假，动作飘忽不定，运动幅度不够大，无法指哪打哪的行业问题，实现从 “视频生成” 到 “演技生成”，从 “动态流畅” 到 “情感表达” 的革命性跨越，标志着 AI 视频生成技术正式从追求 “形似” 进入追求 “神似” 的新纪元，将为内容创作、影视产业、广告营销等领域带来全新升级。然而，当前的原

avatar 2048 AI社区