trw.callbacks.callback_reporting_layer_weights

Module Contents

Classes

CallbackReportingLayerWeights

Report the weight statistics of each layer

Functions

extract_metrics(p: torch.nn.parameter.Parameter)

Attributes

logger

trw.callbacks.callback_reporting_layer_weights.logger

trw.callbacks.callback_reporting_layer_weights.extract_metrics(p: torch.nn.parameter.Parameter)

class trw.callbacks.callback_reporting_layer_weights.CallbackReportingLayerWeights(dataset_name=None, split_name=None, table_name='layer_weights')

Bases: trw.callbacks.callback.Callback

Report the weight statistics of each layer

first_time(self, options, datasets)

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