MXNET
Pywin
这个作者很懒,什么都没留下…
展开
-
mxnet SigmoidBinaryCrossEntropy代码分析
主要对下面这段代码进行一下详细的分析吧# We use the stable formula: max(x, 0) - x * z + log(1 + exp(-abs(x))) loss = F.relu(pred) - pred * label + F.Activation(-F.abs(pred), act_type='softrelu')可能有会疑惑,这不像是交叉熵的公式。这个...原创 2019-12-27 14:56:19 · 616 阅读 · 0 评论 -
mxnet制作.lst和.rec文件
mxnet读数据的其中一种方式就是读取.rec文件,实际上.rec文件的制作流程并不复杂,分为两个步骤:1.制作.lst文件,这里是一种灵活度比较高的方法 上图是一个简单的lst文件,第一列是index列,第二列是标签列,第三列是图片目录列,中间的空隙是制表符'\t'。其中标签列可能是多标签也可能是一个标签,在制作rec文件的时候会略有不同,图片的目录列我认为最好是写一个比...原创 2019-12-09 17:53:24 · 58640 阅读 · 2 评论 -
mxnet如何只训练指定层,其他层参数固定不变?
比如说只更新最后一个dense层的权重:Trainer里面用下面这个而不是所有的model.collect_params()model.collect_params(select='.*dense0')如果想设置为只有dense0层不更新model.collect_params('.*dense0').setattr('grad_req','null')我也是刚了解到这个操作,...原创 2019-12-09 11:45:26 · 867 阅读 · 0 评论 -
linux 安装GPU版本的mxnet
1.先查看自己cuda的版本:cat /usr/local/cuda/version.txt2.建议先卸载之前安装的mxnetpip uninstall mxnetpip uninstall mxnet-cu903.例如cuda的版本是9.0的pip install -i https://pypi.tuna.tsinghua.edu.cn/simple mxnet-cu...原创 2019-11-18 16:13:12 · 59408 阅读 · 0 评论