-
Notifications
You must be signed in to change notification settings - Fork 7
TORCHSCRIPT_CLASSIFIER node for any user-provided .torchscript AI model #234
Conversation
This resolves issue on the CI. Signed-off-by: Julien Jerphanion <[email protected]>
Signed-off-by: Julien Jerphanion <[email protected]>
Signed-off-by: Julien Jerphanion <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This looks like a valuable contribution to me.
To review this PR, I locally updated this branch merging develop
(now used as a base branch), performed a few required changes, and pushed it on my fork for now: https://github.com/jjerphan/nodes/commits/reda-torchscript-classifier/).
Note related to this PR: we might want to take some time to think of reorganizing the AI_ML
section.
I have a few comments and questions.
AI_ML/CLASSIFICATION/TORCHSCRIPT_CLASSIFIER/TORCHSCRIPT_CLASSIFIER_test_.py
Outdated
Show resolved
Hide resolved
AI_ML/CLASSIFICATION/ONE_HOT_ENCODING/ONE_HOT_ENCODING_test_.py
Outdated
Show resolved
Hide resolved
AI_ML/CLASSIFICATION/TORCHSCRIPT_CLASSIFIER/assets/President_Barack_Obama.jpg
Outdated
Show resolved
Hide resolved
AI_ML/CLASSIFICATION/TORCHSCRIPT_CLASSIFIER/assets/mbnet_v3_small.torchscript
Outdated
Show resolved
Hide resolved
…FIER_test_.py Co-authored-by: Julien Jerphanion <[email protected]>
Co-authored-by: Julien Jerphanion <[email protected]>
Co-authored-by: Julien Jerphanion <[email protected]>
…oy-io/nodes into reda-torchscript-classifier
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM! Thank you, @Roulbac.
Just one last side-note.
I let you squash (not to have the intermediary commit for the serialized model) and merge when ready.
Description
Styleguide
Docs
Testing