Я использую Tensorflow 2.0 и пытаюсь написать tf.keras.callbacks.Callback
, который считывает входные и выходные данные моего model
для пакета.
Я ожидал, что смогу переопределить on_batch_end
и получить доступ к model.inputs
и model.outputs
, но это не EagerTensor
со значением, к которому я мог бы получить доступ.Есть ли какой-либо доступ к фактическим значениям тензоров, которые были включены в пакет?
Это имеет много практических применений, таких как вывод этих тензоров в Tensorboard для отладки или сериализация их для других целей.Я знаю, что я мог бы просто запустить всю модель снова, используя model.predict
, но это заставило бы меня запускать каждый вход дважды по сети (и я мог бы также иметь недетерминированный генератор данных).Есть идеи, как этого добиться?