Java 加载自己的数据集时出错。NumClass和InputSplit

Java 加载自己的数据集时出错。NumClass和InputSplit,java,dataset,deeplearning4j,dl4j,Java,Dataset,Deeplearning4j,Dl4j,加载自己的数据集时出现意外异常。执行来自的代码时,只需对神经网络要考虑的样本数进行微小更改,就会产生不支持操作的异常 numClasses = 96 而不是最初的5。拥有 错误: 读卡器输出:96个输出类,但array.size(1)为58 (必须等于1或numclass=96) 版本信息: 深度学习4J 1.0.0-alpha CUDA 8 爪哇8 跟踪: 从日志来看,测试集中似乎有58个示例。所以,也许你在什么地方换了轴。这是我的错!我没有更改新数据集中图像的大小。您确定您所做

加载自己的数据集时出现意外异常。执行来自的代码时,只需对神经网络要考虑的样本数进行微小更改,就会产生不支持操作的异常

numClasses = 96 
而不是最初的5。拥有

错误:

读卡器输出:96个输出类,但array.size(1)为58 (必须等于1或numclass=96)

版本信息:

  • 深度学习4J 1.0.0-alpha
  • CUDA 8
  • 爪哇8
跟踪:


从日志来看,测试集中似乎有58个示例。所以,也许你在什么地方换了轴。

这是我的错!我没有更改新数据集中图像的大小。

您确定您所做的唯一更改是NumClass吗?检查您使用的数据集是否有96个类。@ReinierHernándezÁvila,我确定。只有numclass和数据集。@reinierrenándezÁvila哦,我使用
ZooModel ZooModel=VGG16()而不是
ZooModel ZooModel=VGG16.builder().build()。这可能是错误的原因吗?我最近开始使用ZooModel,所以我不是专业人士,但初始化新网络的方法是:
modelnet=ZooModel.init()。还有更多的文档我有这个消息:
15:05:12830 INFO~trainData 230 15:05:12830 INFO~testData 58 15:05:12838 INFO~numExamples 6011 15:05:12838 INFO~numLabels 96
我决定检查它接收到的数据