add hist

calico · Apr 24, 2024 · 160394b · 160394b
1 parent ac1a250
commit 160394b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
         ]