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

Conform to new torchvision API for specifying pretrained weights #1794

Merged
merged 1 commit into from Jun 14, 2023

Conversation

AdeelH
Copy link
Collaborator

@AdeelH AdeelH commented Jun 14, 2023

Overview

In version 0.13, TorchVision introduced a new API for specifying pretrained weights and added a deprecation warning for the old API. This PR updates the instances of the use of the old API within Raster Vision to the new API.

This is entirely an internal change and does not affect the ModelConfig API, which still accepts the pretrained field.

Checklist

  • Added needs-backport label if PR is bug fix that applies to previous minor release
  • Ran scripts/format_code and committed any changes
  • Documentation updated if needed
  • PR has a name that won't get you publicly shamed for vagueness

Notes

N/A

Testing Instructions

  • There should no longer be deprecation warnings when creating models.

@AdeelH AdeelH marked this pull request as ready for review June 14, 2023 19:33
@codecov
Copy link

codecov bot commented Jun 14, 2023

Codecov Report

❗ No coverage uploaded for pull request base (master@6b44624). Click here to learn what that means.
The diff coverage is 100.00%.

@@            Coverage Diff            @@
##             master    #1794   +/-   ##
=========================================
  Coverage          ?   79.52%           
=========================================
  Files             ?      187           
  Lines             ?     9037           
  Branches          ?        0           
=========================================
  Hits              ?     7187           
  Misses            ?     1850           
  Partials          ?        0           
Impacted Files Coverage Δ
...n/pytorch_learner/classification_learner_config.py 95.38% <100.00%> (ø)
...pytorch_learner/object_detection_learner_config.py 86.66% <100.00%> (ø)
...ision/pytorch_learner/regression_learner_config.py 88.29% <100.00%> (ø)
...ch_learner/semantic_segmentation_learner_config.py 92.13% <100.00%> (ø)

@AdeelH AdeelH merged commit 1be9d21 into azavea:master Jun 14, 2023
2 checks passed
@AdeelH AdeelH deleted the weights branch June 14, 2023 20:41
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.

None yet

1 participant