trw.callbacks.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, exclusions)

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

Attributes

logger

trw.callbacks.callback_zip_sources.logger
trw.callbacks.callback_zip_sources.zip_sources(roots, location, extensions, exclusions)

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

class trw.callbacks.callback_zip_sources.CallbackZipSources(folders_to_record, extensions=('.py', '.sh', '.bat', '.json'), filename='sources.zip', max_width=200, exclusions=('.mypy_cache', '.svn', '.git', '__pycache__'))

Bases: trw.callbacks.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)