trw.train.callback_model_summary

Module Contents

Classes

CallbackModelSummary

Display important characteristics of the model (e.g., FLOPS, number of parameters, layers, shapes)

Functions

model_summary(model, batch, logger)

Attributes

logger

trw.train.callback_model_summary.logger

trw.train.callback_model_summary.model_summary(model, batch, logger)

class trw.train.callback_model_summary.CallbackModelSummary(logger=utilities.log_and_print, dataset_name=None, split_name=None)

Bases: trw.train.callback.Callback

Display important characteristics of the model (e.g., FLOPS, number of parameters, layers, shapes)

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