Skip to content

Commit

Permalink
Bump the torch group in /requirements with 2 updates (#2353)
Browse files Browse the repository at this point in the history
* Bump the torch group in /requirements with 2 updates

Bumps the torch group in /requirements with 2 updates: [torch](https://github.com/pytorch/pytorch) and [torchvision](https://github.com/pytorch/vision).


Updates `torch` from 2.4.1 to 2.5.0
- [Release notes](https://github.com/pytorch/pytorch/releases)
- [Changelog](https://github.com/pytorch/pytorch/blob/main/RELEASE.md)
- [Commits](pytorch/pytorch@v2.4.1...v2.5.0)

Updates `torchvision` from 0.19.1 to 0.20.0
- [Release notes](https://github.com/pytorch/vision/releases)
- [Commits](pytorch/vision@v0.19.1...v0.20.0)

---
updated-dependencies:
- dependency-name: torch
  dependency-type: direct:production
  update-type: version-update:semver-minor
  dependency-group: torch
- dependency-name: torchvision
  dependency-type: direct:production
  update-type: version-update:semver-minor
  dependency-group: torch
...

Signed-off-by: dependabot[bot] <support@github.com>

* Remove unused ignores

---------

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
  • Loading branch information
dependabot[bot] and adamjstewart authored Oct 17, 2024
1 parent ee0e086 commit 3559517
Show file tree
Hide file tree
Showing 5 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion experiments/torchgeo/benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -217,7 +217,7 @@ def main(args: argparse.Namespace) -> None:

criterion = nn.CrossEntropyLoss()
params = model.parameters()
optimizer = optim.SGD(params, lr=0.0001) # type: ignore[attr-defined]
optimizer = optim.SGD(params, lr=0.0001)

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu', args.device)
model = model.to(device)
Expand Down
4 changes: 2 additions & 2 deletions requirements/required.txt
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,6 @@ rtree==1.3.0
segmentation-models-pytorch==0.3.4
shapely==2.0.6
timm==0.9.7
torch==2.4.1
torch==2.5.0
torchmetrics==1.4.3
torchvision==0.19.1
torchvision==0.20.0
2 changes: 1 addition & 1 deletion torchgeo/trainers/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@

import lightning
from lightning.pytorch import LightningModule
from torch.optim import AdamW # type: ignore[attr-defined]
from torch.optim import AdamW
from torch.optim.lr_scheduler import ReduceLROnPlateau


Expand Down
2 changes: 1 addition & 1 deletion torchgeo/trainers/iobench.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
import lightning
import torch
from torch import Tensor
from torch.optim import SGD # type: ignore[attr-defined]
from torch.optim import SGD

from .base import BaseTask

Expand Down
2 changes: 1 addition & 1 deletion torchgeo/trainers/moco.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from lightly.models.utils import deactivate_requires_grad, update_momentum
from lightly.utils.scheduler import cosine_schedule
from torch import Tensor
from torch.optim import SGD, AdamW, Optimizer # type: ignore[attr-defined]
from torch.optim import SGD, AdamW, Optimizer
from torch.optim.lr_scheduler import (
CosineAnnealingLR,
LinearLR,
Expand Down

0 comments on commit 3559517

Please sign in to comment.