Skip to main content


Tools for integrating wandb with Keras.

Keras is a deep learning API for TensorFlow.


class WandbCallback: WandbCallback automatically integrates keras with wandb.

class WandbEvalCallback: Abstract base class to build Keras callbacks for model prediction visualization.

class WandbMetricsLogger: Logger that sends system metrics to W&B.

class WandbModelCheckpoint: A checkpoint that periodically saves a Keras model or model weights.

On this page
Was this page helpful?๐Ÿ‘๐Ÿ‘Ž