trw.callbacks.callback_tensorboard_record_model

Module Contents

Classes

CallbackTensorboardRecordModel

This callback will export the model to tensorboard

Attributes

logger

trw.callbacks.callback_tensorboard_record_model.logger

class trw.callbacks.callback_tensorboard_record_model.CallbackTensorboardRecordModel(dataset_name=None, split_name=None, onnx_folder='onnx')

Bases: trw.callbacks.callback_tensorboard.CallbackTensorboardBased

This callback will export the model to tensorboard

@TODO ONNX is probably adding hooks and are not removed. To be investigated.

__call__(self, options, history, model, losses, outputs, datasets, datasets_infos, callbacks_per_batch, **kwargs)