pytorch/serve
Vedi su GitHub[Suggestion] Loading torch-model-archiver arguments from YAML
Open
#780 aperta il 10 nov 2020
enhancementgood first issuetriaged_wait
Metriche repository
- Star
- (3844 star)
- Metriche merge PR
- (Nessuna PR mergiata in 30 g)
Descrizione
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.