Skip to content

Update qfedavg.py

Update qfedavg.py #536

Workflow file for this run

# This is a basic workflow to help you get started with Actions
name: CI
on:
push:
branches:
- 'master'
- 'feature-test-siqi'
- 'release-v**'
pull_request:
branches:
- 'master'
# A workflow run is made up of one or more jobs that can run sequentially or in parallel
jobs:
# This workflow contains a single job called "build"
build:
# The type of runner that the job will run on
runs-on: ${{ matrix.os }}
strategy:
matrix:
python-version: [3.8] #[ 3.6, 3.7, 3.8 ]
torch-version: [1.7.1] #[1.7.1, 1.8.0]
os: [ubuntu-latest] #[ubuntu-latest, macos-latest]
include:
- os: ubuntu-latest
path: ~/.cache/pip
# - os: macos-latest
# path: ~/Library/Caches/pip
# check https://pypi.org/project/torchvision/ for torch-torchvision compatibility
- torch-version: 1.7.1
torchvision-version: 0.8.2
# - torch-version: 1.8.0
# torchvision-version: 0.9.0
steps:
- uses: actions/checkout@v2
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v2
with:
python-version: ${{ matrix.python-version }}
- name: Cache pip
uses: actions/cache@v2
with:
# This path is depend on os
path: ${{ matrix.path }}
# Look to see if there is a cache hit for the corresponding requirements file
key: ${{ runner.os }}-pip-py${{ matrix.python-version }}-torch==${{ matrix.torch-version }}
restore-keys: |
${{ runner.os }}-pip-
# - name: Install dependencies
# run: pip install -r requirements.txt
- name: Install PyTorch & Torchvision
run: |
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
- name: Install other dependencies
run: |
pip install numpy
# - name: Test with coverage
# run: |
# pip install coverage
# pip install codecov
# coverage run --source=fedlab setup.py test
# - name: Upload Coverage to Codecov
# uses: codecov/codecov-action@v1
# with:
# token: ${{ secrets.CODECOV_TOKEN }}
# name: codecov-umbrella
# fail_ci_if_error: true