我有一个Keras模型,在训练期间需要2个输入(2个输出对应2个损失函数)
2个输入和2个输出成对连接
因此,在推理中,我实际上不需要传递第二个输入,也不需要第二个输出
有没有办法让Keras/tf.Keras使用predict
方法只接受第一个输入并生成第一个输出,而忽略第二个输入和第二个输出
我可以为第二个输入创建一个归零的numpy数组,但我想知道是否有可能减少内存使用或计算
Tensorflow应该能够做到这一点,因为它的图形是惰性的。但凯拉斯有能力做到这一点吗
例如:
# assume second_batch is not needed
second_batch = np.zeros(shape=first_batch.shape)
results = model.predict((first_batch, second_batch))
# i only care about results[0]
# not results[1]
您始终可以使用相同的共享权重构造新的
keras
模型,并指定所需的输入和输出张量输出:
相关问题 更多 >
编程相关推荐