BoundsError: попытка доступа к 2n-элементному UnitRange {Int32} по индексу [2n + 1]

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]

в зависимости от batchsize * 2. На моем локальном компьютере код работает нормально, протестировал его и все работает с меньшими примерами наборов данных. На сервере, где Knet работает нормально, мой код вылетает, хотя.

Любые предложения, как исправить? Уже пробовал разные пакеты, но, похоже, ничего не работает.

Answer 1

Обнаружил ошибку, это был выходной слой с размером 2. Я забыл изменить его на 4, так как сейчас я использую 4 набора данных, а не 2 больше. Спасибо всем, кто занимается расследованием.