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.