trw.train.callback_export_augmentations

Module Contents

Classes

CallbackExportAugmentations

Export samples

Attributes

logger

trw.train.callback_export_augmentations.logger

class trw.train.callback_export_augmentations.CallbackExportAugmentations(nb_samples=10, nb_augmentation=5, dirname='augmentations', split_name=None, uid_name='sample_uid', keep_samples=False)

Bases: trw.train.callback.Callback

Export samples

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