Проверка обновлений стоимости для RBM в Theano

Question

У меня есть следующая тео-функция

train_rbm = theano.function(
    [index],
    cost,
    updates=updates,
    givens={
        x: train_set_x[index * batch_size: (index + 1) * batch_size]
    },
    name='train_rbm'
)

, и затем для каждой эпохи я записываю значение из функции стоимости

for epoch in range(training_epochs):
    mean_cost = []
    for batch_index in range(n_train_batches):
        mean_cost += [train_rbm(batch_index)]

    print('Training epoch %d, cost is ' % epoch, np.mean(mean_cost))

Но во время тренировки я получил

Training epoch 0, cost is  nan   
Training epoch 1, cost is  nan

Это потому, что моя функция Theano не имеет вывода?Как определить его в таком случае?

Если вам нужна ссылка на полный исходный код, я следую этому учебнику