julia - BoundsError：尝试访问索引 [2n+1] 处的 2n 元素 UnitRange{Int32}

Question

我最近在使用 Knet 并在 CNN 上工作。我有一个20000 x 1 x 1 x 200000用于训练的数据集输入

一旦我开始训练progress!(sgd(dtrn,ncycle(n)))，我就会出现以下错误：

BoundsError: attempt to access 2n-element UnitRange{Int32} at index [2n+1]

取决于批大小 * 2。在我的本地计算机上，代码运行良好，经过测试，一切都适用于较小的数据集示例。在 Knet 工作正常的服务器上，我的代码崩溃了。

任何建议如何解决？已经尝试了不同的批量，但似乎没有任何效果。

Answer 1

发现错误，它是大小为 2 的输出层。我忘记将其更改为 4，因为我现在使用 4 个数据集而不是 2。感谢大家调查。