Блог

Я разрабатываю пользовательскую функцию потерь в Keras, def loss(y_true, y_pred) . Но я хочу повторно использовать один из этих тензоров (y_pred) в другой нейронной сети. Я передаю его объекту модели.

Когда я пытаюсь запустить свой код, я получаю сообщение об ошибке: ValueError: When feeding symbolic tensors to a model, we expect the tensors to have a static batch size. Got tensor with shape: (None, 256, 256, 3) .

Я компилирую модель: network.compile(loss=loss, optimizer='sgd')

И запустите эту функцию потерь, используя network.fit(x=random_array, y=np.zeros((BATCH_SIZE, imgdim[0], imgdim[1], imgdim[2])))

Как я могу приспособить y_pred тензор к другой сети?