- 博客(3)
- 问答 (7)
- 收藏
- 关注
RSS订阅原创 ValueError: Expected more than 1 value per channel when training, got input size
原因： BN需要多个数据进行计算平均值等，只有一个数据输入时会产生该错误解决办法：1、考虑是否是训练时最后一个batch刚好只有一个数据，可修改batch size 或者设置DataLoader中drop_last设置为True丢弃最后一个batch的数据2、考虑验证集测试时是否设置model.eval() ，该函数可以保证测试时不计算新的bn层参数...
2021-11-26 13:57:12
991
原创 RuntimeError: CUDA error: device-side assert triggered解决方法
RuntimeError: CUDA error: device-side assert triggered解决方法pytorch训练分割网络出现该问题，注意检查网络输出类别数是否和标签中类别数一致，标签中是否有其他奇怪的像素值，如1.2，10.8等。
2021-11-16 10:41:49
1110
空空如也
PCL点云配准中寻找对应关系
2020-03-22
PCl+cmake配置过程报错，关于Boost
2020-03-10
关于在Cmake下vs2017和vtk的配置问题
2020-02-14
c++ 关于虚函数的返回值问题
2020-02-06
c++ 关于两个vector<int>的初始化问题
2019-11-16
c++ toupper函数和迭代器的报错
2019-10-30
TA创建的收藏夹 TA关注的收藏夹
TA关注的人