- 博客(3)
- 问答 (7)
- 收藏
- 关注
原创 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 1031
原创 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 1115
空空如也
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关注的人