Added MLX as an alternative backend to PyTorch; changed file format for model saving to .safetensors (supported by both MLX and PyTorch). #1716
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: PyTorch CI Tests | |
| permissions: | |
| contents: read | |
| pull-requests: read | |
| on: | |
| push: | |
| branches: [main] | |
| pull_request: | |
| branches: [main] | |
| # Allows you to run this workflow manually from the Actions tab | |
| workflow_dispatch: | |
| jobs: | |
| PyTorch-Tests: | |
| runs-on: ubuntu-latest | |
| 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 pytest tests | |
| uv run plato.py -c configs/MNIST/fedavg_lenet5.toml | |
| uv run plato.py -c configs/MNIST/fedavg_cross_silo_lenet5.toml | |
| uv run plato.py -c configs/HuggingFace/fedavg_opt_lora.toml | |
| uv run plato.py -c configs/MNIST/fedavg_async_lenet5.toml | |
| uv run plato.py -c configs/MNIST/fedavg_lenet5_noniid.toml |