example-emnist.yml

name: Example emnist

on: [push]

jobs:
  train:

    runs-on: ${{ matrix.os }}
    strategy:
      max-parallel: 8
      matrix:
        python-version: [3.7, 3.8, 3.9, '3.10', 3.11, 3.12]
        os: [macos-13, windows-latest, ubuntu-latest]
        include:
          - pytorch-version: 1.4.0
            torchvision-version: 0.5.0
          - pytorch-version: 1.8.0
            torchvision-version: 0.9.0
            python-version: 3.9
          - pytorch-version: 1.11.0
            torchvision-version: 0.12.0
            python-version: '3.10'
          - pytorch-version: 2.0.0
            torchvision-version: 0.15.1
            python-version: 3.11
          - pytorch-version: 2.2.0
            torchvision-version: 0.17.0 
            python-version: 3.12
          - pytorch-option: '+cpu'
          - pytorch-option: ''
            os: macos-13

    steps:
    - uses: actions/checkout@v4
    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v5
      with:
        python-version: ${{ matrix.python-version }}
    - name: Install dependencies
      run: |
        python -m pip install --upgrade pip
        pip install 'numpy<2' -U
        pip install torch==${{ matrix.pytorch-version }}${{ matrix.pytorch-option }} torchvision==${{ matrix.torchvision-version }}${{ matrix.pytorch-option }} -f https://download.pytorch.org/whl/torch_stable.html
        pip install setuptools
        pip install requests
    - name: Install package
      run: python -m pip install .
    - name: Download EMNIST dataset
      run: python examples/emnist/download.py
    - name: Train a model on EMNIST dataset
      run: python examples/emnist/main.py --epochs 1