Skip to content

Latest commit

 

History

History
38 lines (25 loc) · 1.9 KB

README.md

File metadata and controls

38 lines (25 loc) · 1.9 KB

FL Tutorial (UQ CSSE4011)

Federated learning tutorial given as part of the UQ course on Advanced Embedded Systems CSSE4011.
In this tutorial we use Flower and Pytorch frameworks.


Contact : Abdelwahed khamis & Sara Khalifa


Setup

  • Local setup. If you wish to run the notebooks locally, type the following in your terminal.
    conda install -c conda-forge flower
  • Online Notebook. Just click the "Open in Colab" button below and it will launch it in Google Collab. Ensure that you run the cell with the command below to install requirements
    pip install -q flwr[simulation] torch torchvision matplotlib
    • GPU acceleration on Collab : Runtime > Change runtime type > Hardware acclerator: GPU > Save).
    • If you see an error related to GPU availability in one of the following sections, consider switching back to CPU-based execution by setting DEVICE = torch.device("cpu").

Notebooks