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

handle 1d outputs from custom pytorch models #45

Merged
merged 2 commits into from
Jan 19, 2024

Conversation

edknv
Copy link
Contributor

@edknv edknv commented Jan 18, 2024

No description provided.

@edknv edknv force-pushed the feat/custom-pytroch-1d-arr branch from 664928c to 6f03f59 Compare January 19, 2024 18:17
Copy link

Benchmark comparison for 382a6d8 (main) vs 6f03f597 (PR)


---------------------------------------------------------------------------------------------- benchmark: 2 tests ---------------------------------------------------------------------------------------------
Name (time in s)                                  Min                 Max                Mean            StdDev              Median               IQR            Outliers     OPS            Rounds  Iterations
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
test_fiqa_all_MiniLM_L6_v2 (0001_main)        99.2222 (1.0)      105.8137 (1.01)     101.8252 (1.0)      2.6432 (1.23)     101.7152 (1.0)      3.8823 (1.42)          1;0  0.0098 (1.0)           5           1
test_fiqa_all_MiniLM_L6_v2 (0002_6f03f59)     99.2945 (1.00)     104.7682 (1.0)      102.7605 (1.01)     2.1466 (1.0)      103.3234 (1.02)     2.7391 (1.0)           1;0  0.0097 (0.99)          5           1
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

Legend:
  Outliers: 1 Standard Deviation from Mean; 1.5 IQR (InterQuartile Range) from 1st Quartile and 3rd Quartile.
  OPS: Operations Per Second, computed as 1 / Mean

@edknv edknv marked this pull request as ready for review January 19, 2024 19:07
@edknv edknv merged commit 148609d into rapidsai:main Jan 19, 2024
10 checks passed
@edknv edknv deleted the feat/custom-pytroch-1d-arr branch January 19, 2024 19:10
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.

1 participant