Skip to content

Support creating of model bundles from a directory - #251

Merged
yixu34 merged 12 commits into
hmi-releasefrom
yi/support-create-model-bundle-from-dir
Mar 12, 2022
Merged

Support creating of model bundles from a directory#251
yixu34 merged 12 commits into
hmi-releasefrom
yi/support-create-model-bundle-from-dir

Conversation

@yixu34

@yixu34 yixu34 commented Mar 9, 2022 •
edited
Loading

Copy link
Copy Markdown
Member

Allows users to create model bundles from a local zip file. Here's an example invocation that I tested:

model_bundle = deploy_client.create_model_bundle_from_dir(
    model_bundle_name="yi-bundle-dir-test6",
    base_path="/Users/yixu/projects/models/lidar",
    requirements_path="/Users/yixu/projects/models/lidar/requirements.txt",
    env_params={"framework_type": "pytorch", "pytorch_image_tag": "1.7.1-cuda11.0-cudnn8-runtime"}, 
    load_predict_fn_module_path="lidar.segmentation.tasks.load_predict_fn", 
    load_model_fn_module_path="lidar.segmentation.models.decorator.get_segmentation_model"
)

@yixu34
yixu34 requested a review from a team March 11, 2022 02:18
@yixu34 yixu34 self-assigned this Mar 11, 2022
@yixu34
yixu34 marked this pull request as ready for review March 11, 2022 02:19
syandroo
syandroo approved these changes Mar 12, 2022
View reviewed changes

@syandroo syandroo left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

cool!

@yixu34

yixu34 commented Mar 12, 2022

Copy link
Copy Markdown
Member Author

Gonna merge since this seems unrelated (plus it's in a feature branch)
image

@yixu34
yixu34 merged commit b23b2a1 into hmi-release Mar 12, 2022
@yixu34
yixu34 deleted the yi/support-create-model-bundle-from-dir branch March 12, 2022 03:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

1 more reviewer
@syandroo syandroo syandroo approved these changes
Reviewers whose approvals may not affect merge requirements

Assignees

@yixu34 yixu34

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@yixu34 @syandroo