.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "_examples/toy_example.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download__examples_toy_example.py>`
        to download the full example code.

.. rst-class:: sphx-glr-example-title

.. _sphx_glr__examples_toy_example.py:


A FashionMNIST Example
==================

.. GENERATED FROM PYTHON SOURCE LINES 6-57

.. code-block:: Python



    import ray
    from ray import tune
    from ray.tune.stopper import MaximumIterationStopper

    from blades.algorithms.fedavg import FedavgConfig, Fedavg
    from fedlib.trainers import TrainerConfig


    class ExampleFedavgConfig(FedavgConfig):
        def __init__(self, algo_class=None):
            """Initializes a FedavgConfig instance."""
            super().__init__(algo_class=algo_class or ExampleFedavg)

            self.dataset_config = {
                "type": "FashionMNIST",
                "num_clients": 4,
                "train_batch_size": 32,
            }
            self.global_model = "cnn"


    class ExampleFedavg(Fedavg):
        def __init__(self, config=None, logger_creator=None, **kwargs):
            super().__init__(config, logger_creator, **kwargs)

        @classmethod
        def get_default_config(cls) -> TrainerConfig:
            return ExampleFedavgConfig()


    if __name__ == "__main__":
        ray.init()

        config_dict = (
            ExampleFedavgConfig()
            .resources(
                num_gpus_for_driver=0.5,
                num_cpus_for_driver=1,
                num_remote_workers=0,
                num_gpus_per_worker=0.5,
            )
            .to_dict()
        )
        print(config_dict)
        tune.run(
            ExampleFedavg,
            config=config_dict,
            stop=MaximumIterationStopper(100),
        )


.. _sphx_glr_download__examples_toy_example.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: toy_example.ipynb <toy_example.ipynb>`

    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: toy_example.py <toy_example.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: toy_example.zip <toy_example.zip>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_