Fixing a parameter

miladsade96 · Dec 21, 2021 · 9ddf9f7 · 9ddf9f7
1 parent 4b3d190
commit 9ddf9f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -44,7 +44,7 @@
 
 # Defining callback objects
 tensorboard_callback = TensorBoard(log_dir="./tb_logs", histogram_freq=1, write_graph=True,
-                                   write_images=False, update_freq="epoch")
+                                   write_images=True, update_freq="epoch")
 
 # Compiling the model
 model.compile(optimizer=Adam(learning_rate=args.learning_rate), loss=total_loss, metrics=metrics)