我目前正在使用Mxnet和C++ Symbol API训练一个卷积神经网络。该网络包含一些批处理归一化层,其中包含四个参数NDArray。其中两个参数,moving_mean和moving_variance参数应该在训练期间的每一批中更新。 我猜测,由于executor的前向传递的布尔值设置为true,因此它将自动更新新参数。此外,由于没有为这两个NDArray计算梯度,因为它不是“可学习”的参数,所以我无法通过常规
enter image description here我在一个文件夹中有3000个用于训练和测试的图像,我还有label.csv文件中的图像标签,其中包含五个类别。谁能帮我把这个数据集分成训练数据和测试数据,这样我就可以使用卷积神经网络对图像进行分类了。将csv和图像链接后,我的数据集看起来如下图所示。