Я создал пользовательскую оценку keras, расширив tf.keras.Model в tf2. В этом классе я расширил методы call, loss и fit. Теперь я хочу записать резюме и сгенерировать график тензорной доски для обученной модели. Не уверен, как реализовать то же самое с пользовательской моделью.