trw.train.callback_export_best_history

Module Contents

Classes

CallbackExportBestHistory

Export the best value of the history and epoch for each metric in a single file

Attributes

logger

trw.train.callback_export_best_history.logger

class trw.train.callback_export_best_history.CallbackExportBestHistory(filename='best_history.txt', metric_to_discard=[])

Bases: trw.train.callback.Callback

Export the best value of the history and epoch for each metric in a single file

This can be useful to accurately get the best value of a metric and in particular at which step it occurred.

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