pytorch/serve
Voir sur GitHub[Suggestion] Loading torch-model-archiver arguments from YAML
Open
#780 ouverte le 10 nov. 2020
enhancementgood first issuetriaged_wait
Métriques du dépôt
- Stars
- (3 844 stars)
- Métriques de merge PR
- (Aucune PR mergée en 30 j)
Description
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.