trw.callbacks.callback_reporting_model_summary

Module Contents

Classes

CallbackReportingModelSummary

Defines a callback function that may be called before training, during training, after training

Functions

input_shape(i, root=True)
model_summary_base(model, batch)
export_table(options, table_name, batch, table_role, clear_existing_data, base_name='', table_preamble='', commit=True)
html_list(items, header=None)

Attributes

logger

trw.callbacks.callback_reporting_model_summary.logger

trw.callbacks.callback_reporting_model_summary.input_shape(i, root=True)

trw.callbacks.callback_reporting_model_summary.model_summary_base(model, batch)

trw.callbacks.callback_reporting_model_summary.export_table(options, table_name, batch, table_role, clear_existing_data, base_name='', table_preamble='', commit=True)

trw.callbacks.callback_reporting_model_summary.html_list(items, header=None)

class trw.callbacks.callback_reporting_model_summary.CallbackReportingModelSummary(dataset_name=None, split_name=None)

Bases: trw.callbacks.callback.Callback

Defines a callback function that may be called before training, during training, after training

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