Skip to content
Snippets Groups Projects
Commit 34456fe2 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

path for tflogs

parent 2209e855
No related branches found
No related tags found
1 merge request!20Mccn trainer
Pipeline #26853 passed
......@@ -119,7 +119,7 @@ def main(user_input=None):
# train the network
if hasattr(configuration, 'network'):
trainer = MCCNNTrainer(configuration.network, batch_size=batch_size, use_gpu=use_gpu, adapted_layers=adapted_layers, adapt_reference_channel=adapt_reference_channel, verbosity_level=verbosity_level)
trainer = MCCNNTrainer(configuration.network, batch_size=batch_size, use_gpu=use_gpu, adapted_layers=adapted_layers, adapt_reference_channel=adapt_reference_channel, verbosity_level=verbosity_level,tf_logdir=output_dir+'/tf_logs')
trainer.train(dataloader, n_epochs=epochs, learning_rate=learning_rate, output_dir=output_dir, model=model)
else:
logger.error("Please provide a network in your configuration file !")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment