深度学习Deep Learning训练错误
Posted 一厘阳光
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了深度学习Deep Learning训练错误相关的知识,希望对你有一定的参考价值。
错误提示:Check failed: axis_index < num_axes() (1 vs. 1) axis 1 out of range for 1
这里目前出现在实践过程中,有两方面会引起此错误,若各位大佬有其他的发现,多多交流哈。
第一种:
解决办法:在Layer层的设置上,去看一下是不是你的label和data放反了,如下图:
第二种:
网络结构中,使用ImageData作为网络数据输入,其中包含独立的concat层,会出现此错误。
解决方案:将concat层注掉即可。
以上是关于深度学习Deep Learning训练错误的主要内容,如果未能解决你的问题,请参考以下文章
深度学习模型训练技巧 Tips for Deep Learning
Deep learning:四十三(用Hessian Free方法训练Deep Network)
Deep Learning(深度学习)之Deep Learning的常用模型或者方法
翻译: Deep Learning深度学习平台Hugging Face 开源代码和技术构建训练和部署 ML 模型