openfl.callbacks.MemoryProfiler#
- class openfl.callbacks.MemoryProfiler(log_dir='./logs/')[source]#
Profile memory usage of the current process at the end of each round.
- Parameters:
log_dir (str)
- log_dir#
If set, writes logs as lines of JSON.
Methods
__init__([log_dir])on_experiment_begin([logs])Callback function to be executed at the beginning of an experiment.
on_experiment_end([logs])Callback function to be executed at the end of an experiment.
on_round_begin(round_num[, logs])Callback function to be executed at the beginning of a round.
on_round_end(round_num[, logs])Callback function to be executed at the end of a round.
set_params(params)set_tensor_db(tensor_db)