pytorch/serve
GitHub で見る[Suggestion] Loading torch-model-archiver arguments from YAML
Open
#780 opened on 2020年11月10日
enhancementgood first issuetriaged_wait
Repository metrics
- Stars
- (3,844 stars)
- PR merge metrics
- (30d に merged 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.