plato-learn 1.1__tar.gz

plato_learn-1.1/.bashrc

@@ -0,0 +1,100 @@
+# ~/.bashrc: executed by bash(1) for non-login shells.
+# see /usr/share/doc/bash/examples/startup-files (in the package bash-doc)
+# for examples
+
+# If not running interactively, don't do anything
+[ -z "$PS1" ] && return
+
+# don't put duplicate lines in the history. See bash(1) for more options
+# ... or force ignoredups and ignorespace
+HISTCONTROL=ignoredups:ignorespace
+
+# append to the history file, don't overwrite it
+shopt -s histappend
+
+# for setting history length see HISTSIZE and HISTFILESIZE in bash(1)
+HISTSIZE=1000
+HISTFILESIZE=2000
+
+# check the window size after each command and, if necessary,
+# update the values of LINES and COLUMNS.
+shopt -s checkwinsize
+
+# make less more friendly for non-text input files, see lesspipe(1)
+[ -x /usr/bin/lesspipe ] && eval "$(SHELL=/bin/sh lesspipe)"
+
+# set variable identifying the chroot you work in (used in the prompt below)
+if [ -z "$debian_chroot" ] && [ -r /etc/debian_chroot ]; then
+    debian_chroot=$(cat /etc/debian_chroot)
+fi
+
+# set a fancy prompt (non-color, unless we know we "want" color)
+case "$TERM" in
+    xterm-color) color_prompt=yes;;
+esac
+
+# uncomment for a colored prompt, if the terminal has the capability; turned
+# off by default to not distract the user: the focus in a terminal window
+# should be on the output of commands, not on the prompt
+#force_color_prompt=yes
+
+if [ -n "$force_color_prompt" ]; then
+    if [ -x /usr/bin/tput ] && tput setaf 1 >&/dev/null; then
+	# We have color support; assume it's compliant with Ecma-48
+	# (ISO/IEC-6429). (Lack of such support is extremely rare, and such
+	# a case would tend to support setf rather than setaf.)
+	color_prompt=yes
+    else
+	color_prompt=
+    fi
+fi
+
+if [ "$color_prompt" = yes ]; then
+    PS1='${debian_chroot:+($debian_chroot)}\[\033[01;32m\]\u@\h\[\033[00m\]:\[\033[01;34m\]\w\[\033[00m\]\$ '
+else
+    PS1='${debian_chroot:+($debian_chroot)}\u@\h:\w\$ '
+fi
+unset color_prompt force_color_prompt
+
+# If this is an xterm set the title to user@host:dir
+case "$TERM" in
+xterm*|rxvt*)
+    PS1="\[\e]0;${debian_chroot:+($debian_chroot)}\u@\h: \w\a\]$PS1"
+    ;;
+*)
+    ;;
+esac
+
+# enable color support of ls and also add handy aliases
+if [ -x /usr/bin/dircolors ]; then
+    test -r ~/.dircolors && eval "$(dircolors -b ~/.dircolors)" || eval "$(dircolors -b)"
+    alias ls='ls --color=auto'
+    #alias dir='dir --color=auto'
+    #alias vdir='vdir --color=auto'
+
+    alias grep='grep --color=auto'
+    alias fgrep='fgrep --color=auto'
+    alias egrep='egrep --color=auto'
+fi
+
+# some more ls aliases
+alias ll='ls -alF'
+alias la='ls -A'
+alias l='ls -CF'
+
+# Alias definitions.
+# You may want to put all your additions into a separate file like
+# ~/.bash_aliases, instead of adding them here directly.
+# See /usr/share/doc/bash-doc/examples in the bash-doc package.
+
+if [ -f ~/.bash_aliases ]; then
+    . ~/.bash_aliases
+fi
+
+# enable programmable completion features (you don't need to enable
+# this, if it's already enabled in /etc/bash.bashrc and /etc/profile
+# sources /etc/bash.bashrc).
+#if [ -f /etc/bash_completion ] && ! shopt -oq posix; then
+#    . /etc/bash_completion
+#fi
+

plato_learn-1.1/.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,31 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: "[BUG]"
+labels: bug
+assignees: baochunli
+
+---
+
+**Describe the bug**
+A clear and concise description of what the bug is.
+
+**To Reproduce**
+Steps to reproduce the behavior:
+1. Go to '...'
+2. Click on '....'
+3. Scroll down to '....'
+4. See error
+
+**Expected behavior**
+A clear and concise description of what you expected to happen.
+
+**Screenshots**
+If applicable, add screenshots to help explain your problem.
+
+**OS environment (please complete the following information):**
+ - OS: [e.g. macOS, Ubuntu Linux]
+ - Version [e.g. 10.15.7]
+
+**Additional context**
+Add any other context about the problem here.

plato_learn-1.1/.github/ISSUE_TEMPLATE/feature_request.md

@@ -0,0 +1,20 @@
+---
+name: Feature request
+about: Suggest an idea for this project
+title: "[FR]"
+labels: enhancement
+assignees: baochunli
+
+---
+
+**Is your feature request related to a problem? Please describe.**
+A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
+
+**Describe the solution you'd like**
+A clear and concise description of what you want to happen.
+
+**Describe alternatives you've considered**
+A clear and concise description of any alternative solutions or features you've considered.
+
+**Additional context**
+Add any other context or screenshots about the feature request here.

plato_learn-1.1/.github/ISSUE_TEMPLATE/request-for-comments.md

@@ -0,0 +1,11 @@
+---
+name: Request for comments
+about: Architectural changes, milestone setting, project management, community rules,
+  and general discussion about Plato 
+title: "[RFC]"
+labels: rfc
+assignees: baochunli
+
+---
+
+

plato_learn-1.1/.github/pull_request_template.md

@@ -0,0 +1,25 @@
+<!--- Provide a general summary of your changes in the Title above -->
+
+## Description
+<!--- Describe your changes in detail -->
+<!--- Describe motivation and context -->
+<!--- Why is this change required? What problem does it solve? -->
+<!--- If it fixes an open issue, please link to the issue here. -->
+
+## How has this been tested?
+<!--- Please describe in detail how you tested your changes. -->
+<!--- Include details of your testing environment, tests ran to see how -->
+<!--- your change affects other areas of the code, etc. -->
+
+## Types of changes
+<!--- What types of changes does your code introduce? Put an `x` in all the boxes that apply: -->
+- [ ] Bug fix (non-breaking change which fixes an issue) Fixes #
+- [ ] New feature (non-breaking change which adds functionality)
+- [ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
+
+## Checklist:
+<!--- Go over all the following points, and put an `x` in all the boxes that apply. -->
+<!--- If you're unsure about any of these, don't hesitate to ask. We're here to help! -->
+- [ ] My code has been formatted using Black and checked using PyLint.
+- [ ] My change requires a change to the documentation.
+- [ ] I have updated the documentation accordingly.

plato_learn-1.1/.github/workflows/configs/fedavg_yolov8.yml

@@ -0,0 +1,179 @@
+clients:
+    # Type
+    type: simple
+
+    # The total number of clients
+    total_clients: 1
+
+    # The number of clients selected in each round
+    per_round: 1
+
+    # Should the clients compute test accuracy locally?
+    do_test: true
+
+    random_seed: 1
+
+server:
+    address: 127.0.0.1
+    port: 8000
+
+data:
+    # The training and testing dataset
+    datasource: YOLOv8
+    data_params: configs/yolov8/data/coco128.yaml
+
+    # Where the dataset is located
+    data_path: data/COCO
+    train_path: data/COCO/coco128/images/train2017/
+    test_path: data/COCO/coco128/images/train2017/
+
+    # download command/URL (optional)
+    download_urls:
+        [
+            "https://ultralytics.com/assets/coco128.zip",
+        ]
+
+    # number of training examples
+    num_train_examples: 128
+
+    # number of testing examples
+    num_test_examples: 128
+
+    # number of classes
+    num_classes: 80
+
+    # image size
+    image_size: 640
+
+    # class names
+    classes:
+        [
+            "person",
+            "bicycle",
+            "car",
+            "motorcycle",
+            "airplane",
+            "bus",
+            "train",
+            "truck",
+            "boat",
+            "traffic light",
+            "fire hydrant",
+            "stop sign",
+            "parking meter",
+            "bench",
+            "bird",
+            "cat",
+            "dog",
+            "horse",
+            "sheep",
+            "cow",
+            "elephant",
+            "bear",
+            "zebra",
+            "giraffe",
+            "backpack",
+            "umbrella",
+            "handbag",
+            "tie",
+            "suitcase",
+            "frisbee",
+            "skis",
+            "snowboard",
+            "sports ball",
+            "kite",
+            "baseball bat",
+            "baseball glove",
+            "skateboard",
+            "surfboard",
+            "tennis racket",
+            "bottle",
+            "wine glass",
+            "cup",
+            "fork",
+            "knife",
+            "spoon",
+            "bowl",
+            "banana",
+            "apple",
+            "sandwich",
+            "orange",
+            "broccoli",
+            "carrot",
+            "hot dog",
+            "pizza",
+            "donut",
+            "cake",
+            "chair",
+            "couch",
+            "potted plant",
+            "bed",
+            "dining table",
+            "toilet",
+            "tv",
+            "laptop",
+            "mouse",
+            "remote",
+            "keyboard",
+            "cell phone",
+            "microwave",
+            "oven",
+            "toaster",
+            "sink",
+            "refrigerator",
+            "book",
+            "clock",
+            "vase",
+            "scissors",
+            "teddy bear",
+            "hair drier",
+            "toothbrush",
+        ]
+
+    # Number of samples in each partition
+    partition_size: 128
+
+    # IID or non-IID?
+    sampler: iid
+
+trainer:
+    # The type of the trainer
+    type: yolov8
+
+    # test function
+    testtype: map
+
+    # The maximum number of training rounds
+    rounds: 1
+
+    # The maximum number of clients running concurrently
+    max_concurrency: 3
+
+    # The target accuracy
+    target_accuracy: 0.99
+
+    # Number of epoches for local training in each communication round
+    epochs: 1
+    batch_size: 4
+    optimizer: Adam
+    linear_lr: false
+
+    # The machine learning model
+    model_name: yolov8
+
+algorithm:
+    # Aggregation algorithm
+    type: fedavg
+
+parameters:
+    grid_size: 32
+
+    model:
+        type: yolov8n.pt
+        cfg: yolov8n.yaml
+        num_classes: 80
+
+    optimizer:
+        lr: 0.01
+        momentum: 0.9
+        weight_decay: 0.0

plato_learn-1.1/.github/workflows/delete_workflow_runs.yml

@@ -0,0 +1,27 @@
+# This workflow will delete old workflow runs, on a schedule or manually triggered.
+
+name: Delete old workflow runs
+
+on:
+  workflow_dispatch:
+    inputs:
+      days:
+        description: 'Number of days.'
+        required: true
+        default: 30
+      minimum_runs:
+        description: 'The minimum runs to keep for each workflow.'
+        required: true
+        default: 6
+
+jobs:
+  del_runs:
+    runs-on: ubuntu-20.04
+    steps:
+      - name: Delete workflow runs
+        uses: Mattraks/delete-workflow-runs@v2
+        with:
+          token: ${{ github.token }}
+          repository: ${{ github.repository }}
+          retain_days: ${{ github.event.inputs.days }}
+          keep_minimum_runs: ${{ github.event.inputs.minimum_runs }}

plato_learn-1.1/.github/workflows/gpu_tests.yml.disabled

@@ -0,0 +1,36 @@
+name: Continuous Integration Tests
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+  # Allows you to run this workflow manually from the Actions tab
+  workflow_dispatch:
+    branches: [main]
+
+jobs:
+  Tests:
+    runs-on: ubuntu-20.04
+
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Install CUDA
+        env:
+          cuda: "11.3"
+        run: bash ./scripts/actions/install_cuda_ubuntu.sh
+        shell: bash
+
+      - name: Training workloads
+        run: |
+          mkdir ./models/pretrained
+          wget https://kubeedge.obs.cn-north-1.myhuaweicloud.com/model/yolov5.pth
+          mv yolov5.pth ./models/pretrained
+          uv run plato.py --config=configs/MNIST/fedavg_lenet5.yml
+          uv run plato.py --config=configs/MNIST/mistnet_lenet5.yml
+          uv run plato.py --config=.github/workflows/configs/fedavg_yolov5.yml
+          uv run plato.py --config=.github/workflows/configs/mistnet_yolov5.yml

plato_learn-1.1/.github/workflows/pypi_publish.yml

@@ -0,0 +1,32 @@
+# This workflow will upload a Python Package using uv when a release is created.
+
+name: Upload the plato-learn Python Package to PyPI
+
+on:
+    release:
+        types:
+            - created
+
+jobs:
+    build_and_deploy:
+        runs-on: ubuntu-latest
+
+        steps:
+            - name: Checkout repository
+              uses: actions/checkout@v5
+
+            - name: Install uv
+              uses: astral-sh/setup-uv@v3
+              with:
+                  version: "latest"
+
+            - name: Set up Python
+              uses: actions/setup-python@v5
+              with:
+                  python-version: "3.13"
+
+            - name: Build package
+              run: uv build
+
+            - name: Publish package to PyPI
+              run: uv publish --token ${{ secrets.PYPI_API_TOKEN }}

plato_learn-1.1/.github/workflows/pytorch_tests.yml

@@ -0,0 +1,32 @@
+name: PyTorch CI Tests
+
+on:
+    push:
+        branches: [main]
+    pull_request:
+        branches: [main]
+    # Allows you to run this workflow manually from the Actions tab
+    workflow_dispatch:
+        branches: [main]
+
+jobs:
+    PyTorch-Tests:
+        runs-on: ubuntu-20.04
+
+        steps:
+            - name: Checkout
+              uses: actions/checkout@v5
+              with:
+                  submodules: recursive
+
+            - name: Install uv
+              uses: astral-sh/setup-uv@v4
+
+            - name: Training workloads
+              shell: bash -l {0}
+              run: |
+                  uv run plato.py --config=configs/MNIST/fedavg_lenet5.yml
+                  uv run plato.py --config=configs/MNIST/fedavg_cross_silo_lenet5.yml
+                  uv run plato.py --config=configs/MNIST/fedavg_async_lenet5.yml
+                  uv run plato.py --config=configs/MNIST/fedavg_lenet5_noniid.yml
+                  uv run plato.py --config=configs/MNIST/mistnet_lenet5.yml

plato_learn-1.1/.gitignore

@@ -0,0 +1,163 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+.pypirc
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# Visual Studio Code
+.vscode/
+
+# Configuration files
+/*.yml
+
+# Data files
+data/
+tests/data/
+
+# Models and checkpoints
+pretrained/
+/models/
+logs/
+checkpoints/
+*.pth
+
+# Result files
+results/
+
+# macOS
+.DS_Store
+
+# Plain text files in the root working directory are most likely temporary
+/*.txt
+
+# Sphinx documentation
+docs/_build/
+
+# Homomorphic encryption
+.ckks_context/
+yolov8n.pt
+
+# uv
+uv.lock

plato_learn-1.1/.gitmodules

@@ -0,0 +1,12 @@
+[submodule "plato/models/dvit"]
+	path = plato/models/dvit
+	url = https://github.com/zhoudaquan/dvit_repo
+[submodule "plato/models/t2tvit"]
+	path = plato/models/t2tvit
+	url = https://github.com/yitu-opensource/T2T-ViT
+[submodule "examples/pfedrlnas/VIT/nasvit_wrapper/NASViT"]
+	path = examples/pfedrlnas/VIT/nasvit_wrapper/NASViT
+	url = https://github.com/facebookresearch/NASViT
+[submodule "examples/controlnet_split_learning/ControlNet"]
+	path = examples/split_learning/controlnet_split_learning/ControlNet
+	url = https://github.com/lllyasviel/ControlNet

plato_learn-1.1/.python-version

@@ -0,0 +1 @@
+3.13

plato_learn-1.1/Dockerfile

@@ -0,0 +1,20 @@
+# Download base image from NVIDIA's Docker Hub
+FROM nvidia/cuda:13.0.1-cudnn-devel-ubuntu24.04
+LABEL maintainer="Baochun Li"
+
+WORKDIR /root/plato
+COPY . /root/plato/
+
+RUN apt-get update \
+    && apt-get install -y libgomp1 \
+    && apt-get install -y curl \
+    && curl -LsSf https://astral.sh/uv/install.sh | sh
+
+# Set environment variables for uv before using it
+ENV PATH="/root/.local/bin:$PATH"
+
+# Set additional environment variables for uv
+ENV UV_SYSTEM_PYTHON=1
+
+# Set default command to use uv run
+CMD ["/bin/bash"]