Skip to content

Commit f3529f4

Browse files
author
Seppo Enarvi
committed
Merge branch 'master' into generic-weight-averaging
2 parents 5a69057 + 69d8fa9 commit f3529f4

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

46 files changed

+149
-159
lines changed

.azure/gpu-tests-fabric.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,13 +60,13 @@ jobs:
6060
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.1"
6161
PACKAGE_NAME: "fabric"
6262
"Fabric | latest":
63-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
64-
PACKAGE_NAME: "fabric"
65-
"Fabric | future":
6663
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.7-cuda12.6.3"
6764
PACKAGE_NAME: "fabric"
65+
#"Fabric | future":
66+
# image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.7-cuda12.6.3"
67+
# PACKAGE_NAME: "fabric"
6868
"Lightning | latest":
69-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
69+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.7-cuda12.6.3"
7070
PACKAGE_NAME: "lightning"
7171
workspace:
7272
clean: all

.azure/gpu-tests-pytorch.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,13 +53,13 @@ jobs:
5353
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.1"
5454
PACKAGE_NAME: "pytorch"
5555
"PyTorch | latest":
56-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
57-
PACKAGE_NAME: "pytorch"
58-
"PyTorch | future":
5956
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.7-cuda12.6.3"
6057
PACKAGE_NAME: "pytorch"
58+
#"PyTorch | future":
59+
# image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.7-cuda12.6.3"
60+
# PACKAGE_NAME: "pytorch"
6161
"Lightning | latest":
62-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
62+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.7-cuda12.6.3"
6363
PACKAGE_NAME: "lightning"
6464
pool: lit-rtx-3090
6565
variables:

.github/CODEOWNERS

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,3 +18,6 @@
1818
/README.md @williamfalcon @lantiga
1919
/src/pytorch_lightning/__about__.py @williamfalcon @lantiga @borda
2020
/src/lightning_fabric/__about__.py @williamfalcon @lantiga @borda
21+
22+
/src/lightning/fabric/loggers @williamfalcon
23+
/src/lightning/pytorch/loggers @williamfalcon

.github/checkgroup.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,8 @@ subprojects:
134134
- "build-pl (3.11, 2.3, 12.1.1)"
135135
- "build-pl (3.11, 2.4, 12.1.1)"
136136
- "build-pl (3.12, 2.5, 12.1.1)"
137-
- "build-pl (3.12, 2.6, 12.4.1, true)"
137+
- "build-pl (3.12, 2.6, 12.4.1)"
138+
- "build-pl (3.12, 2.7, 12.6.3, true)"
138139

139140
# SECTION: lightning_fabric
140141

.github/workflows/_legacy-checkpoints.yml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ defaults:
5151

5252
jobs:
5353
create-legacy-ckpts:
54-
runs-on: ubuntu-20.04
54+
runs-on: ubuntu-22.04
5555
outputs:
5656
pl-version: ${{ steps.decide-version.outputs.pl-version }}
5757
steps:
@@ -113,6 +113,7 @@ jobs:
113113

114114
- run: pip install -r requirements/ci.txt
115115
- name: Upload checkpoints to S3
116+
if: secrets.AWS_REGION != ''
116117
working-directory: ${{ env.LEGACY_FOLDER }}
117118
env:
118119
AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY }}
@@ -124,7 +125,7 @@ jobs:
124125
aws s3 cp $AWS_RUN checkpoints.zip s3://pl-public-data/legacy/ --acl public-read
125126
126127
add-ckpt-test:
127-
runs-on: ubuntu-20.04
128+
runs-on: ubuntu-22.04
128129
if: inputs.create_pr
129130
needs: create-legacy-ckpts
130131
env:

.github/workflows/call-clear-cache.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,18 +23,18 @@ on:
2323
jobs:
2424
cron-clear:
2525
if: github.event_name == 'schedule' || github.event_name == 'pull_request'
26-
uses: Lightning-AI/utilities/.github/workflows/[email protected].2
26+
uses: Lightning-AI/utilities/.github/workflows/[email protected].3
2727
with:
28-
scripts-ref: v0.14.2
28+
scripts-ref: v0.14.3
2929
dry-run: ${{ github.event_name == 'pull_request' }}
3030
pattern: "latest|docs"
3131
age-days: 7
3232

3333
direct-clear:
3434
if: github.event_name == 'workflow_dispatch' || github.event_name == 'pull_request'
35-
uses: Lightning-AI/utilities/.github/workflows/[email protected].2
35+
uses: Lightning-AI/utilities/.github/workflows/[email protected].3
3636
with:
37-
scripts-ref: v0.14.2
37+
scripts-ref: v0.14.3
3838
dry-run: ${{ github.event_name == 'pull_request' }}
3939
pattern: ${{ inputs.pattern || 'pypi_wheels' }} # setting str in case of PR / debugging
4040
age-days: ${{ fromJSON(inputs.age-days) || 0 }} # setting 0 in case of PR / debugging

.github/workflows/ci-check-md-links.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ on:
1414

1515
jobs:
1616
check-md-links:
17-
uses: Lightning-AI/utilities/.github/workflows/[email protected].2
17+
uses: Lightning-AI/utilities/.github/workflows/[email protected].3
1818
with:
1919
config-file: ".github/markdown-links-config.json"
2020
base-branch: "master"

.github/workflows/ci-schema.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ on:
88

99
jobs:
1010
check:
11-
uses: Lightning-AI/utilities/.github/workflows/[email protected].2
11+
uses: Lightning-AI/utilities/.github/workflows/[email protected].3
1212
with:
1313
# skip azure due to the wrong schema file by MSFT
1414
# https://github.com/Lightning-AI/lightning-flash/pull/1455#issuecomment-1244793607

.github/workflows/ci-tests-fabric.yml

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -41,13 +41,13 @@ jobs:
4141
include:
4242
# only run PyTorch latest
4343
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
44-
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
44+
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
4545
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
4646
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2.2" }
47-
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2.2" }
47+
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2.2" }
4848
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2.2" }
4949
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.3" }
50-
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.3" }
50+
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.3" }
5151
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.3" }
5252
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.12.7", pytorch-version: "2.4.1" }
5353
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.12.7", pytorch-version: "2.4.1" }
@@ -61,11 +61,11 @@ jobs:
6161
- { os: "windows-2022", pkg-name: "fabric", python-version: "3.12.7", pytorch-version: "2.6" }
6262
# "oldest" versions tests, only on minimum Python
6363
- { os: "macOS-14", pkg-name: "fabric", pytorch-version: "2.1", requires: "oldest" }
64-
- { os: "ubuntu-20.04", pkg-name: "fabric", pytorch-version: "2.1", requires: "oldest" }
64+
- { os: "ubuntu-22.04", pkg-name: "fabric", pytorch-version: "2.1", requires: "oldest" }
6565
- { os: "windows-2022", pkg-name: "fabric", pytorch-version: "2.1", requires: "oldest" }
6666
# "fabric" installs the standalone package
6767
- { os: "macOS-14", pkg-name: "fabric", python-version: "3.10", pytorch-version: "2.5" }
68-
- { os: "ubuntu-20.04", pkg-name: "fabric", python-version: "3.10", pytorch-version: "2.5" }
68+
- { os: "ubuntu-22.04", pkg-name: "fabric", python-version: "3.10", pytorch-version: "2.5" }
6969
- { os: "windows-2022", pkg-name: "fabric", python-version: "3.10", pytorch-version: "2.5" }
7070
# adding recently cut Torch 2.7 - FUTURE
7171
- { os: "macOS-14", pkg-name: "fabric", python-version: "3.12", pytorch-version: "2.7" }
@@ -159,7 +159,8 @@ jobs:
159159
run: |
160160
echo $GITHUB_RUN_ID
161161
python -m coverage run --source ${{ env.COVERAGE_SCOPE }} \
162-
-m pytest -v --timeout=30 --durations=50 --random-order-seed=$GITHUB_RUN_ID
162+
-m pytest -v --timeout=30 --durations=50 --random-order-seed=$GITHUB_RUN_ID \
163+
--junitxml=junit.xml -o junit_family=legacy # NOTE: for Codecov's test results
163164
164165
- name: Statistics
165166
if: success()

.github/workflows/ci-tests-pytorch.yml

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -45,13 +45,13 @@ jobs:
4545
include:
4646
# only run PyTorch latest
4747
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
48-
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
48+
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
4949
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
5050
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2.2" }
51-
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2.2" }
51+
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2.2" }
5252
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2.2" }
5353
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.3" }
54-
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.3" }
54+
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.3" }
5555
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.3" }
5656
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.12.7", pytorch-version: "2.4.1" }
5757
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.12.7", pytorch-version: "2.4.1" }
@@ -65,11 +65,11 @@ jobs:
6565
- { os: "windows-2022", pkg-name: "pytorch", python-version: "3.12.7", pytorch-version: "2.6" }
6666
# "oldest" versions tests, only on minimum Python
6767
- { os: "macOS-14", pkg-name: "pytorch", pytorch-version: "2.1", requires: "oldest" }
68-
- { os: "ubuntu-20.04", pkg-name: "pytorch", pytorch-version: "2.1", requires: "oldest" }
68+
- { os: "ubuntu-22.04", pkg-name: "pytorch", pytorch-version: "2.1", requires: "oldest" }
6969
- { os: "windows-2022", pkg-name: "pytorch", pytorch-version: "2.1", requires: "oldest" }
7070
# "pytorch" installs the standalone package
7171
- { os: "macOS-14", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "2.5" }
72-
- { os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "2.5" }
72+
- { os: "ubuntu-22.04", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "2.5" }
7373
- { os: "windows-2022", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "2.5" }
7474
# adding recently cut Torch 2.7 - FUTURE
7575
- { os: "macOS-14", pkg-name: "pytorch", python-version: "3.12", pytorch-version: "2.7" }
@@ -196,7 +196,8 @@ jobs:
196196
run: |
197197
echo $GITHUB_RUN_ID
198198
python -m coverage run --source ${{ env.COVERAGE_SCOPE }} \
199-
-m pytest . -v --timeout=60 --durations=50 --random-order-seed=$GITHUB_RUN_ID
199+
-m pytest . -v --timeout=60 --durations=50 --random-order-seed=$GITHUB_RUN_ID \
200+
--junitxml=junit.xml -o junit_family=legacy # NOTE: for Codecov's test results
200201
201202
- name: Statistics
202203
if: success()
@@ -205,6 +206,12 @@ jobs:
205206
coverage report
206207
coverage xml
207208
209+
- name: Upload test results to Codecov
210+
if: ${{ !cancelled() }} # Run even if tests fail
211+
uses: codecov/test-results-action@v1
212+
with:
213+
token: ${{ secrets.CODECOV_TOKEN }}
214+
208215
- name: Upload coverage to Codecov
209216
uses: codecov/codecov-action@v5
210217
# see: https://github.com/actions/toolkit/issues/399

.github/workflows/code-checks.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ defaults:
2626

2727
jobs:
2828
mypy:
29-
runs-on: ubuntu-20.04
29+
runs-on: ubuntu-22.04
3030
steps:
3131
- uses: actions/checkout@v4
3232
- uses: actions/setup-python@v5

.github/workflows/docker-build.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,8 @@ jobs:
4848
- { python_version: "3.11", pytorch_version: "2.3", cuda_version: "12.1.1" }
4949
- { python_version: "3.11", pytorch_version: "2.4", cuda_version: "12.1.1" }
5050
- { python_version: "3.12", pytorch_version: "2.5", cuda_version: "12.1.1" }
51-
- { python_version: "3.12", pytorch_version: "2.6", cuda_version: "12.4.1", latest: "true" }
51+
- { python_version: "3.12", pytorch_version: "2.6", cuda_version: "12.4.1" }
52+
- { python_version: "3.12", pytorch_version: "2.7", cuda_version: "12.6.3", latest: "true" }
5253
steps:
5354
- uses: actions/checkout@v4
5455
with:

.github/workflows/docs-build.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ env:
5353
jobs:
5454
docs-make:
5555
if: github.event.pull_request.draft == false
56-
runs-on: ubuntu-20.04
56+
runs-on: ubuntu-22.04
5757
strategy:
5858
fail-fast: false
5959
matrix:
@@ -72,7 +72,7 @@ jobs:
7272
lfs: ${{ matrix.pkg-name == 'pytorch' }}
7373
- uses: actions/setup-python@v5
7474
with:
75-
python-version: "3.9"
75+
python-version: "3.10"
7676

7777
- name: List notebooks
7878
if: ${{ matrix.pkg-name == 'pytorch' }}
@@ -149,7 +149,7 @@ jobs:
149149
deploy-docs:
150150
needs: docs-make
151151
if: github.repository_owner == 'Lightning-AI' && github.event_name != 'pull_request'
152-
runs-on: ubuntu-20.04
152+
runs-on: ubuntu-22.04
153153
strategy:
154154
fail-fast: false
155155
matrix:

.github/workflows/docs-tutorials.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ defaults:
1616

1717
jobs:
1818
docs-update:
19-
runs-on: ubuntu-20.04
19+
runs-on: ubuntu-22.04
2020
steps:
2121
- uses: actions/checkout@v4
2222
with:

.pre-commit-config.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ repos:
5151
- id: detect-private-key
5252

5353
- repo: https://github.com/codespell-project/codespell
54-
rev: v2.3.0
54+
rev: v2.4.1
5555
hooks:
5656
- id: codespell
5757
additional_dependencies: [tomli]
@@ -70,7 +70,7 @@ repos:
7070
- id: sphinx-lint
7171

7272
- repo: https://github.com/astral-sh/ruff-pre-commit
73-
rev: v0.8.6
73+
rev: v0.11.4
7474
hooks:
7575
# try to fix what is possible
7676
- id: ruff
@@ -81,7 +81,7 @@ repos:
8181
- id: ruff
8282

8383
- repo: https://github.com/executablebooks/mdformat
84-
rev: 0.7.21
84+
rev: 0.7.22
8585
hooks:
8686
- id: mdformat
8787
additional_dependencies:

_notebooks

docs/source-fabric/conf.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -292,6 +292,8 @@
292292
("py:.*", "torch_xla.*"),
293293
("py:class", "transformer_engine.*"),
294294
("py:class", "bitsandbytes.*"),
295+
# loggers
296+
('py:class', 'tensorboardX.SummaryWriter'), # todo: this is unexpected as the imports locally works
295297
]
296298

297299
# -- Options for todo extension ----------------------------------------------

docs/source-pytorch/conf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -377,6 +377,7 @@ def _load_py_module(name: str, location: str) -> ModuleType:
377377
# missing in generated API
378378
("py:exc", "MisconfigurationException"),
379379
# TODO: generated list of all existing ATM, need to be fixed
380+
('py:class', 'tensorboardX.SummaryWriter'),
380381
("py:class", "AveragedModel"),
381382
("py:class", "CometExperiment"),
382383
("py:meth", "DataModule.__init__"),

docs/source-pytorch/visualize/loggers.rst

Lines changed: 0 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -54,37 +54,3 @@ Track and Visualize Experiments
5454

5555
</div>
5656
</div>
57-
58-
.. _mlflow_logger:
59-
60-
MLflow Logger
61-
-------------
62-
63-
The MLflow logger in PyTorch Lightning now includes a `checkpoint_path_prefix` parameter. This parameter allows you to prefix the checkpoint artifact's path when logging checkpoints as artifacts.
64-
65-
Example usage:
66-
67-
.. code-block:: python
68-
69-
import lightning as L
70-
from lightning.pytorch.loggers import MLFlowLogger
71-
72-
mlf_logger = MLFlowLogger(
73-
experiment_name="lightning_logs",
74-
tracking_uri="file:./ml-runs",
75-
checkpoint_path_prefix="my_prefix"
76-
)
77-
trainer = L.Trainer(logger=mlf_logger)
78-
79-
# Your LightningModule definition
80-
class LitModel(L.LightningModule):
81-
def training_step(self, batch, batch_idx):
82-
# example
83-
self.logger.experiment.whatever_ml_flow_supports(...)
84-
85-
def any_lightning_module_function_or_hook(self):
86-
self.logger.experiment.whatever_ml_flow_supports(...)
87-
88-
# Train your model
89-
model = LitModel()
90-
trainer.fit(model)

requirements/ci.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,4 @@ twine ==6.0.1
55
importlib-metadata <8.0.0
66
wget
77
pkginfo ==1.12.0
8-
packaging <24.2
8+
packaging <25.1

requirements/docs.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ sphinx-rtd-dark-mode
1717
sphinxcontrib-video ==0.2.0
1818
jinja2 <3.2.0
1919

20-
lightning-utilities >=0.11.1, <0.12.0
20+
lightning-utilities >=0.11.1, <0.15.0
2121

2222
# installed from S3 location and fetched in advance
2323
lai-sphinx-theme

requirements/doctests.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
pytest ==7.4.0
2-
pytest-doctestplus ==1.0.0
1+
pytest ==8.3.5
2+
pytest-doctestplus ==1.4.0

requirements/fabric/base.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
22
# in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
33

4-
torch >=2.1.0, <2.6.0
4+
torch >=2.1.0, <2.7.0
55
fsspec[http] >=2022.5.0, <2024.4.0
6-
packaging >=20.0, <=23.1
6+
packaging >=20.0, <=25.0
77
typing-extensions >=4.4.0, <4.11.0
8-
lightning-utilities >=0.10.0, <0.12.0
8+
lightning-utilities >=0.10.0, <0.15.0

requirements/fabric/examples.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
22
# in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
33

4-
torchvision >=0.16.0, <0.21.0
5-
torchmetrics >=0.10.0, <1.5.0
6-
lightning-utilities >=0.8.0, <0.12.0
4+
torchvision >=0.16.0, <0.22.0
5+
torchmetrics >=0.10.0, <1.8.0
6+
lightning-utilities >=0.8.0, <0.15.0

0 commit comments

Comments
 (0)