Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feature/upgrade pytorch #134

Merged
merged 14 commits into from
Mar 4, 2025
Merged

Feature/upgrade pytorch #134

merged 14 commits into from
Mar 4, 2025

Conversation

lorenzomammana
Copy link
Contributor

@lorenzomammana lorenzomammana commented Feb 26, 2025

Summary

Aim of this PR is to finalize the upgrade of pytorch from 2.1.2 to 2.4.1.
Alongside this pytorch lightning has been upgraded to 2.4.0.
Multiple packages have been updated based on security issues coming from dependabot.

A detailed description of the changes is given in the changelog:

[2.3.0]

Updated

  • Upgrade torch from 2.1.2+cu121 to 2.4.1+cu121
  • Upgrade pytorch lightning to 2.4.0
  • Upgrade onnxruntime-gpu to 1.20.0
  • Upgrade anomalib to support the update lightning version
  • Upgrade problematic packages notified by dependabot
  • Deprecate python 3.9 support
  • Update mypy, ruff and pylint to the latest available versions
  • Apply pre-commits on all files based on the latest versions of the tools

Fixed

  • Increase atol for mixed precision model export to avoid errors on efficientnet_b0 export
  • Increase atol and rtol for model export tests to avoid issues on gpu testing

Type of Change

  • Breaking change (fix or feature that would cause existing functionality to not work as expected)

Checklist

Please confirm that the following tasks have been completed:

  • I have tested my changes locally and they work as expected. (Please describe the tests you performed.)
  • I have added unit tests for my changes, or updated existing tests if necessary.
  • I have updated the documentation, if applicable.
  • I have installed pre-commit and run locally for my code changes.

@lorenzomammana lorenzomammana marked this pull request as ready for review February 26, 2025 14:22
@lorenzomammana lorenzomammana merged commit 3eacb61 into main Mar 4, 2025
1 check passed
@lorenzomammana lorenzomammana deleted the feature/upgrade-pytorch branch March 4, 2025 11:06
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants