trw.train.callback_data_summary

Module Contents

Classes

CallbackDataSummary

Summarizes the data (min value, max value, number of batches, shapes) for each split of each dataset

Attributes

logger

trw.train.callback_data_summary.logger

class trw.train.callback_data_summary.CallbackDataSummary(logger=utilities.log_and_print, collect_stats=True)

Bases: trw.train.callback.Callback

Summarizes the data (min value, max value, number of batches, shapes) for each split of each dataset

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