Add Learning Rate Scheduler to neurallamConfig #359
Workflow file for this run
This file contains 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
# cicd workflow for running tests with pytest | |
# needs to first install pdm, then install torch cpu manually and then install the package | |
# then run the tests | |
name: test (pip install, cpu) | |
on: [push, pull_request] | |
jobs: | |
tests: | |
runs-on: ubuntu-latest | |
steps: | |
- name: Checkout | |
uses: actions/checkout@v2 | |
- name: Install torch (CPU) | |
run: | | |
# torch version pinned to 2.5.1 for now so that we avoid the | |
# downgrade (and then GPU version install) when actually installing | |
# neural-lam | |
python -m pip install "torch==2.5.1" --index-url https://download.pytorch.org/whl/cpu | |
- name: Install package (including dev dependencies) | |
run: | | |
python -m pip install ".[dev]" | |
- name: Print and check torch version | |
run: | | |
python -c "import torch; print(torch.__version__)" | |
python -c "import torch; assert torch.__version__.endswith('+cpu')" | |
- name: Load cache data | |
uses: actions/cache/restore@v4 | |
with: | |
path: tests/datastore_examples/npyfilesmeps/meps_example_reduced.tar.gz | |
key: ${{ runner.os }}-meps-reduced-example-data-v0.3.0 | |
restore-keys: | | |
${{ runner.os }}-meps-reduced-example-data-v0.3.0 | |
- name: Run tests | |
run: | | |
python -m pytest -vv -s | |
- name: Save cache data | |
uses: actions/cache/save@v4 | |
with: | |
path: tests/datastore_examples/npyfilesmeps/meps_example_reduced.tar.gz | |
key: ${{ runner.os }}-meps-reduced-example-data-v0.3.0 |