trw.train.callback_save_last_model

Module Contents

Classes

CallbackSaveLastModel

When the training is finished, save the full model and result

Attributes

logger

trw.train.callback_save_last_model.logger

class trw.train.callback_save_last_model.CallbackSaveLastModel(model_name='last')

Bases: trw.train.callback.Callback

When the training is finished, save the full model and result

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