openfl.callbacks.MemoryProfiler
===============================

.. currentmodule:: openfl.callbacks

.. autoclass:: MemoryProfiler

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~MemoryProfiler.__init__
      ~MemoryProfiler.on_experiment_begin
      ~MemoryProfiler.on_experiment_end
      ~MemoryProfiler.on_round_begin
      ~MemoryProfiler.on_round_end
      ~MemoryProfiler.on_task_begin
      ~MemoryProfiler.on_task_end
      ~MemoryProfiler.set_params
      ~MemoryProfiler.set_tensor_db