Установить тензор из пользовательской функции потерь Keras в другую сеть - PullRequest
4 голосов
/ 10 апреля 2019

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

Когда я пытаюсь запустить свой код, я получаю ошибку: 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 тензор в другую сеть?

...