trw.callbacks.callback_debug_processes

Module Contents

Classes

CallbackDebugProcesses

Defines a callback function that may be called before training, during training, after training

Functions

log_process(process)
log_all_tree(pid=None)
_collect_data(main_process, filename, frequency_seconds, abort_event)

Attributes

psutil
logger

trw.callbacks.callback_debug_processes.psutil

trw.callbacks.callback_debug_processes.log_process(process)

trw.callbacks.callback_debug_processes.log_all_tree(pid=None)

trw.callbacks.callback_debug_processes._collect_data(main_process, filename, frequency_seconds, abort_event)

trw.callbacks.callback_debug_processes.logger

class trw.callbacks.callback_debug_processes.CallbackDebugProcesses(filename='process_stack_dumps', frequency_seconds=10.0, timeout=10.0, delayed_init=True)

Bases: trw.callbacks.callback.Callback

Defines a callback function that may be called before training, during training, after training

_init(self, root='')

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

close(self)

__del__(self)