pytorch/serve

[Suggestion] Loading torch-model-archiver arguments from YAML

Open

#780 opened on Nov 10, 2020

View on GitHub
 (6 comments) (1 reaction) (1 assignee)Java (790 forks)batch import
enhancementgood first issuetriaged_wait

Repository metrics

Stars
 (3,844 stars)
PR merge metrics
 (No merged PRs in 30d)

Description

Is your feature request related to a problem? Please describe.

torch-model-archiver takes in a lot of params that makes it a hefty command line argument

Describe the solution

Allowing all the param values to be parsed from a YAML file when torch-model-archiver is used with a --yaml flag. It will also help automating creation of multiple .mar from different configurations.

Describe alternatives solution

I've tried calling torch-model-archiver from a shell script with variable params and that works file as well. But I just wanted to bounce this idea off the community to see if it is something people could find useful. I'd be happy to work on it.

Contributor guide

Research direction
Implement YAML file parsing for torch model archiver command line arguments using a library like PyYAML.
Tech stack
python
Domain
clidevops
Issue type
Feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
3
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
1-2 days
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Active
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Clear
Prerequisites
PythonYAML
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
60