From 160394bbf99f7c14dcad98a749e322e51444849f Mon Sep 17 00:00:00 2001
From: lruizcalico <lruiz@calicolabs.com>
Date: Wed, 24 Apr 2024 09:36:31 -0700
Subject: [PATCH] add hist

---
 src/baskerville/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/baskerville/trainer.py b/src/baskerville/trainer.py
index 74e6362..7ce6758 100644
--- a/src/baskerville/trainer.py
+++ b/src/baskerville/trainer.py
@@ -190,7 +190,7 @@ def fit_keras(self, seqnn_model):
 
         callbacks = [
             early_stop,
-            tf.keras.callbacks.TensorBoard(self.log_dir),
+            tf.keras.callbacks.TensorBoard(self.log_dir, histogram_freq=1),
             tf.keras.callbacks.ModelCheckpoint("%s/model_check.h5" % self.out_dir),
             save_best,
         ]