niftynet.engine.handler_tensorboard module

This module implements a TensorBoard log writer.

class TensorBoardLogger(model_dir=None, initial_iter=0, tensorboard_every_n=0, **_unused)

Bases: object

This class handles iteration events to log summaries to the TensorBoard log.

init_writer(_sender, **_unused_msg)

Initialise summary writers.

Parameters:

• _sender –
• msg –

Returns:

read_tensorboard_op(sender, **msg)

Get TensorBoard summary_op from application at the beginning of each iteration.

Parameters:

• sender – a niftynet.application instance
• msg – should contain an IterationMessage instance

write_tensorboard(_sender, **msg)

Write to tensorboard when received the iteration finished signal.

Parameters:

• _sender –
• msg –