pytorch/serve

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

Open

#780 创建于 2020年11月10日

在 GitHub 查看
 (6 评论) (1 反应) (1 负责人)Java (790 fork)batch import
enhancementgood first issuetriaged_wait

仓库指标

Star
 (3,844 star)
PR 合并指标
 (30 天内没有已合并 PR)

描述

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.

贡献者指南