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.

貢獻者指南