RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton

深度学习维度不匹配问题

努力成为大牛吧

8943人浏览 · 2023-03-01 20:43:48

努力成为大牛吧 · 2023-03-01 20:43:48 发布

RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton

这个错误主要是因为，数据的数量不能整除batch size的数量，改正方法有两种

将batch size改称可以被数据量整除即可。
将dataloader 中的 drop_last改称False也可。

2048 AI社区

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

更多推荐

【机器学习】24 Markov chain Monte Carlo (MCMC) inference

本章介绍了马尔可夫链蒙特卡洛(MCMC)方法，重点探讨了Gibbs采样和Metropolis-Hastings算法两种核心方法，并分析其应用场景、收敛性及优化策略。内容涵盖Gibbs采样在Ising模型、高斯混合模型中的应用，Metropolis-Hastings算法原理及变体，以及收敛诊断、辅助变量MCMC、退火方法和边缘似然估计等技术。章节还涉及并行计算、自适应MCMC等前沿主题，为贝叶斯推断