/keras_callback_extns

Callbacks for keras to write out weight matrices to image files and collect run_metadata from tensorflow backends

Primary LanguagePython

keras_callback_extns

Callbacks for keras:

  • WeightWriter to write out the weights of Conv2D and Dense layers as images
  • TFRunMetaData to write out compute time and memory usage of a tensorflow backend

Examples

example code is included in the project root