在尝试使用mxnet支持使用h2o deepwater训练多类分类的lenet模型时,我收到以下错误:
加载H2O mxnet绑定。找到CUDA_HOME或CUDA_PATH ……
从错误日志和文档中可以看出 mxnet.sym.Convolution 你的数据需要在 [batch, channels, height, width] 格式。但是,您的数据看起来只包含两个维度(基于此日志: mxnet data input shape: (32,100) )。重新格式化数据,甚至包括大小为1的两个维度,使得输入形状为(1,1,32,100)应解决此问题。
mxnet.sym.Convolution
[batch, channels, height, width]
mxnet data input shape: (32,100)