Druckverlust beim Training in Keras mit Tensorflow Backend

Question

Ich trainiere mein Deep neurales Netzwerk auf Keras (TF-Backend). Ich möchte nur den ersten Verlust während des Trainings DNN drucken. Ich möchte nur sicherstellen, dass meine Initialisierung korrekt ist und so brauche ich den anfänglichen Verlust, der von der DNN berechnet wird, nachdem ich den ersten Vorwärtsdurchlauf gemacht habe.

Keras Callback ermöglicht uns, Verlust nach jeder Epoche zu bestimmen. Ich will es nach dem ersten Trainingsschritt.

Answer 1

Sie können einen benutzerdefinierten Rückruf erstellen mit on_batch_end

https://keras.io/callbacks/