trw.train.callback_zip_sources

Module Contents

Classes

CallbackZipSources

Record important info relative to the training such as the sources & configuration info

Functions

zip_sources(roots, location, extensions)	Zip the content of roots folders with given extensions to a location
default_extensions()

Attributes

logger

trw.train.callback_zip_sources.logger

trw.train.callback_zip_sources.zip_sources(roots, location, extensions)

Zip the content of roots folders with given extensions to a location

trw.train.callback_zip_sources.default_extensions()

class trw.train.callback_zip_sources.CallbackZipSources(folders_to_record, extensions=default_extensions(), filename='sources.zip', max_width=200)

Bases: trw.train.callback.Callback

Record important info relative to the training such as the sources & configuration info

This is to make sure a result can always be easily reproduced. Any configuration option can be safely appended in options[‘runtime’]

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