pytorch/serve
Auf GitHub ansehen[Suggestion] Loading torch-model-archiver arguments from YAML
Open
#780 geöffnet am 10. Nov. 2020
enhancementgood first issuetriaged_wait
Repository-Metriken
- Stars
- (3.844 Stars)
- PR-Merge-Metriken
- (Keine gemergten PRs in 30 T)
Beschreibung
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.