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

Add a tutorial notebook for ONNX inference #2062

Merged
merged 3 commits into from Feb 15, 2024
Merged

Conversation

AdeelH
Copy link
Collaborator

@AdeelH AdeelH commented Feb 15, 2024

Overview

This PR adds a new tutorial notebook demonstrating the use of an ONNX model for inference with RV.

Checklist

  • Added unit tests, if applicable
  • Updated documentation, if applicable
  • Added needs-backport label if the change should be back-ported to the previous release
  • PR has a name that won't get you publicly shamed for vagueness

Notes

The previously installed onnxruntime-gpu version was incompatible with the CUDA version (12) installed in the docker image. This PR bumps the version to onnxruntime-gpu==1.17, installed with --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/, as recommended in the onnxruntime docs.

Testing Instructions

  • Check if tutorial looks good in docs build.

@AdeelH AdeelH force-pushed the onnx-nb branch 2 times, most recently from 94f053b to de4d1bf Compare February 15, 2024 15:14
Copy link

codecov bot commented Feb 15, 2024

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (1157b44) 89.65% compared to head (f9162a7) 89.65%.

Additional details and impacted files
@@           Coverage Diff           @@
##           master    #2062   +/-   ##
=======================================
  Coverage   89.65%   89.65%           
=======================================
  Files         197      197           
  Lines        9762     9763    +1     
=======================================
+ Hits         8752     8753    +1     
  Misses       1010     1010           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@AdeelH AdeelH marked this pull request as ready for review February 15, 2024 16:41
@AdeelH AdeelH merged commit 22a794f into azavea:master Feb 15, 2024
2 checks passed
@AdeelH AdeelH deleted the onnx-nb branch February 15, 2024 16:42
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