- 博客(4)
- 资源 (3)
- 收藏
- 关注
RSS订阅原创 caffe在别人修改的层中，加入了个全连接层，并且设置为共享参数，训练出的模型中只有该层的名称，没有具体的tensor
问题描述：在caffe中，做的是强化学习image caption，想要实现在网络中多加一个全连接层（一共三个net要加，训练baseline的net，训练强化学习的scstnet，以及两者共用的decode_net）。在训练baseline的时候加入了全连接层。网络可以训练出加入那层的参数，并且decode的时候也能产生结果。但是在训练强化学习的scstnet训练能够训练处模型，不会报错。但是在...
2019-08-20 11:28:02
649
原创 caffe，SCST强化学习网络中，在修改网络的时候显示报错，调用的（共享）参数维度不匹配（笔记）
问题如下：I0807 22:23:04.734895 29440 net.cpp:465] Sharing parameters 'predict_param_0' owned by layer 'beam', param index 7F0807 22:23:04.734915 29440 net.cpp:484] Check failed: this_blob->shape() ...
2019-08-07 23:10:55
325
原创 ubuntu18.04关机重启后NVIDIA-SMI has failed 问题解决
问题描述：注：以下解决办法只针对电脑更新重启后导致内核改变（升级）的问题。如果内核能匹配显卡驱动但是还是无法连接那么建议重新安装驱动。电脑重启打开nvidia-smi出现：NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver第一步 为了验证是否属于内核改变或者升级...
2019-08-06 15:32:21
2726
原创 笔记
layer { name: "concat1_t3" type: "Concat" bottom: "lstm0_hidden3" bottom: "fc8_3" bottom: "lstm1_hidden2" bottom: "lstm1_hidden0" top: "concat1_t3"}layer { name: "reshape_t3" type...
2019-08-06 10:40:49
102
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人