- 博客(4)
- 收藏
- 关注
RSS订阅原创 torch冻结训练中bn问题
今天帮同事修改模型冻结训练中bug，记录一下 模型冻结训练中我们采用的冻结方式为 freeze = ['', ] # parameter names to freeze (full or partial) if any(freeze): for k, v in model.named_parameters(): if any(x in k for x in freeze): print('freezing %s' %
2022-02-08 18:37:46
2064
原创 python中argsort函数
caffe中进行分类程序编写时，总是遇到argsort函数。于是做一个验证，记录一下。可以看输出该函数的作用是将a中的元素从小到大排列，提取其对应的index(索引)，a[0]=2在数组里排第2，所以讲索引号0放在第二位。还有如下情况，同理，...
2018-07-07 13:14:42
2099
原创 caffe 学习
一、C++里整型除了有小数是向下取整,忽略小数pool层有ceil和卷积不一样https://github.com/BVLC/caffe/blob/master/src/caffe/layers/pooling_layer.cpppooled_height_ = static_cast<int>(ceil(static_cast<float>(height_ + 2 * p...
2018-07-07 13:02:20
299
原创 ubuntu16.04装caffe配置过程的坑
用了将近一周的时间安装caffe，本人小白一个，也是刚接触深度学习，最近在安装的过程中踩了好多个坑，安装的过程中参考了很多大神的博客，可能是每台机器不一样的缘故，在配置的过程中出现各种报错。本人是在双显卡笔记本安装。 在编译caffe时没有报错，在进行mnist测试时出错 libcudart.so.8.0 cannot open shared object file: No
2018-01-28 15:50:33
238
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人