您的位置:首页 > 其它

发现了cntk 2.0 rnn方面的一个bug

2017-04-12 17:30 453 查看
 SimpleNetworkBuilder里面设置SIMPLERNN RnnType. 会有一个崩溃

要改正需要修改相关代码

SimpleNetworkBuilder.cpp:174

ComputationNetworkPtr SimpleNetworkBuilder::BuildRNNFromDescription()

{

ComputationNetworkBuilder builder(*m_net);

if (m_net->GetTotalNumberOfNodes() < 1) // not built yet

{

unsigned long randomSeed = 1;
size_t numHiddenLayers = m_layerSizes.size() - 2;

size_t numRecurrentLayers = m_recurrentLayers.size();

ComputationNodePtr input, w, b, u, pastValue, output, label, prior;

//add by ljh
if (m_sparse_input)
input = builder.CreateSparseInputNode(L"features", m_layerSizes[0]);
else
input = builder.CreateInputNode(L"features", m_layerSizes[0]);

//input = builder.CreateSparseInputNode(L"features", m_layerSizes[0]);


这是我在github上的回答
https://github.com/Microsoft/CNTK/issues/151
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  CNTK bug 深度学习