- 博客(4)
- 收藏
- 关注
RSS订阅原创 解决自定义batch_norm训练时报错：TypeError: cannot assign ‘torch.cuda.FloatTensor‘ as parameter ‘running_mean‘
其原因正如报错中所提到的，从batch_norm()函数中返回的是一个torch.cuda.FloatTensor类型的running_mean，不能直接赋值给Parameter类型，所以我们在此处可以做一下类型转换，即可消除这个报错，我训练下来应该没啥大问题，如果有错请大佬指正！我做了一些改动，将running_mean和running_var设置为了Parameter，如果直接赋值为tensor类型变量的话，是不会保存这两个参数的，会造成很多不便，但同时也不会报题目中的错误。
2023-07-18 15:33:25
646
1
原创 手势识别（肤色检测）重心计算以及重心变化趋势计算
代码的主体思想就是定义一个类似于FIFO的存储器，每隔十帧图像，会将一个重心数据存入存储器2中，同时存储器0使用存储器1的数据进行覆盖，存储器1使用存储器2的内容进行覆盖，这样就可以做到实时检测，而不用固定采集一定数量帧图像之后再进行检测。其中分子只剩下坐标的相加，分母其实代表图中白色像素块的个数，此时公式就比较利于硬件实现，而且利用分母的计算，我们可以进行一些肤色检测的抗干扰设计，例如。好了，本文到此就结束了，写这篇文章的主要目的还是记录一下几个星期以来的成果，大佬勿喷。（某些原因就不展示了QAQ）
2023-06-20 19:27:00
227
1
原创 HLS:csim由于内存不足而导致仿真失败——SIGSEGV
在其他博客中看到出现此错误最可能原因是数组访问越界，或者内存不足，经过检查之后（将其余全部代码注释，仅剩声明数组代码，仍然csim失败）访问越界被排除，以下是我设置的输入参数以及开辟的数组，实验发现一旦输入维度增大则cim直接失败。同时ap_uint所占用的空间大小为1字节，经过计算，在我设置的输入参数下，开辟的数组已经远远超过此范围，通过调整堆栈空间空间可解决此问题。将此贴作为错误记录。
2023-03-31 11:59:20
688
3
原创 仿真通过AXI_lite接口读写寄存器时axi_awready信号无法拉高的一种原因
仿真通过AXI_lite接口读写寄存器时axi_awready信号无法拉高的一种原因
2022-09-12 17:15:51
2644
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人