From 6d5f5c0c3e043053b58addf9274066f7ec9d91aa Mon Sep 17 00:00:00 2001 From: clement Date: Mon, 18 Nov 2019 19:17:21 +0100 Subject: [PATCH] Removed Tensorboard because call clashed with requirements --- transfer_learning.py | 15 +-------------- 1 file changed, 1 insertion(+), 14 deletions(-) diff --git a/transfer_learning.py b/transfer_learning.py index e91a31a..9d3681d 100644 --- a/transfer_learning.py +++ b/transfer_learning.py @@ -415,21 +415,8 @@ def train_simpler_inception_based_model(dataset_path, save_weights_only=False, mode='max') - # write logs to ./logs for Tensorboard visualization - tensorboard = TensorBoard(log_dir='./logs', - histogram_freq=0, - batch_size=32, - write_graph=True, - write_grads=False, - write_images=False, - embeddings_freq=0, - embeddings_layer_names=None, - embeddings_metadata=None, - embeddings_data=None, - update_freq='epoch') - # callbacks - cb = [save_on_improve, tensorboard] + cb = [save_on_improve] # loss is categorical since we are classifying # if a pre-trained model was specified, we are fine tuning and need to take the custom learning rate into account